{ "model_type": "roberta", "architectures": [ "RobertaForSequenceClassification" ], "pre_weights": [ { "name": "roberta.embeddings.position_embeddings.weight" }, { "name": "roberta.embeddings.word_embeddings.weight" }, { "name": "roberta.embeddings.token_type_embeddings.weight" }, { "name": "roberta.embeddings.LayerNorm.weight" }, { "name": "roberta.embeddings.LayerNorm.bias" }, { "name": "roberta.embeddings.position_ids", "optional": true, "force_dtype": "int64" } ], "post_weights": [ { "name": "classifier.dense.weight" }, { "name": "classifier.dense.bias" }, { "name": "classifier.out_proj.weight" }, { "name": "classifier.out_proj.bias" } ], "num_layers_config_key": "num_hidden_layers", "layer_templates": { "weights": [ { "name": "roberta.encoder.layer.${layer_index}.attention.output.dense.weight" }, { "name": "roberta.encoder.layer.${layer_index}.attention.output.dense.bias" }, { "name": "roberta.encoder.layer.${layer_index}.attention.output.LayerNorm.weight" }, { "name": "roberta.encoder.layer.${layer_index}.attention.output.LayerNorm.bias" }, { "name": "roberta.encoder.layer.${layer_index}.attention.self.query.weight" }, { "name": "roberta.encoder.layer.${layer_index}.attention.self.query.bias" }, { "name": "roberta.encoder.layer.${layer_index}.attention.self.key.weight" }, { "name": "roberta.encoder.layer.${layer_index}.attention.self.key.bias" }, { "name": "roberta.encoder.layer.${layer_index}.attention.self.value.weight" }, { "name": "roberta.encoder.layer.${layer_index}.attention.self.value.bias" }, { "name": "roberta.encoder.layer.${layer_index}.intermediate.dense.weight" }, { "name": "roberta.encoder.layer.${layer_index}.intermediate.dense.bias" }, { "name": "roberta.encoder.layer.${layer_index}.output.dense.weight" }, { "name": "roberta.encoder.layer.${layer_index}.output.dense.bias" }, { "name": "roberta.encoder.layer.${layer_index}.output.LayerNorm.weight" }, { "name": "roberta.encoder.layer.${layer_index}.output.LayerNorm.bias" } ] } }