cerebras.modelzoo.tools.checkpoint_converters.roberta#

Classes

ConfigConverter_Roberta_HF_CS18

ConfigConverter_Roberta_HF_CS21

CS 2.1 config is the same as CS 2.0.

Converter_RobertaPretrainModel_HF_CS

Converter_RobertaPretrainModel_HF_CS18

Converter_RobertaPretrainModel_WithoutOptionalModel_HF_CS21

previous

cerebras.modelzoo.tools.checkpoint_converters.registry.get_cs_model_wrapper_class

next

cerebras.modelzoo.tools.checkpoint_converters.roberta.ConfigConverter_Roberta_HF_CS18