keys_map.json

{
  "pos_embedding": "model.pos_embedding",
  "embedding.weight": "model.embedding.weight",
  "transformer_encoder.layers.0.self_attn.in_proj_weight": "model.transformer_encoder.layers.0.self_attn.in_proj_weight",
  "transformer_encoder.layers.0.self_attn.in_proj_bias": "model.transformer_encoder.layers.0.self_attn.in_proj_bias",
  "transformer_encoder.layers.0.self_attn.out_proj.weight": "model.transformer_encoder.layers.0.self_attn.out_proj.weight",
  "transformer_encoder.layers.0.self_attn.out_proj.bias": "model.transformer_encoder.layers.0.self_attn.out_proj.bias",
  "transformer_encoder.layers.0.linear1.weight": "model.transformer_encoder.layers.0.linear1.weight",
  "transformer_encoder.layers.0.linear1.bias": "model.transformer_encoder.layers.0.linear1.bias",
  "transformer_encoder.layers.0.linear2.weight": "model.transformer_encoder.layers.0.linear2.weight",
  "transformer_encoder.layers.0.linear2.bias": "model.transformer_encoder.layers.0.linear2.bias",
  "transformer_encoder.layers.0.norm1.weight": "model.transformer_encoder.layers.0.norm1.weight",
  "transformer_encoder.layers.0.norm1.bias": "model.transformer_encoder.layers.0.norm1.bias",
  "transformer_encoder.layers.0.norm2.weight": "model.transformer_encoder.layers.0.norm2.weight",
  "transformer_encoder.layers.0.norm2.bias": "model.transformer_encoder.layers.0.norm2.bias",
  "gru.weight_ih": "model.gru.weight_ih",
  "gru.weight_hh": "model.gru.weight_hh",
  "gru.bias_ih": "model.gru.bias_ih",
  "gru.bias_hh": "model.gru.bias_hh",
  "lstm.weight_ih": "model.lstm.weight_ih",
  "lstm.weight_hh": "model.lstm.weight_hh",
  "lstm.bias_ih": "model.lstm.bias_ih",
  "lstm.bias_hh": "model.lstm.bias_hh",
  "rnn.weight_ih": "model.rnn.weight_ih",
  "rnn.weight_hh": "model.rnn.weight_hh",
  "rnn.bias_ih": "model.rnn.bias_ih",
  "rnn.bias_hh": "model.rnn.bias_hh",
  "cognitive_module.0.weight": "model.cognitive_module.0.weight",
  "cognitive_module.0.bias": "model.cognitive_module.0.bias",
  "cognitive_module.2.weight": "model.cognitive_module.2.weight",
  "cognitive_module.2.bias": "model.cognitive_module.2.bias",
  "emotional_module.0.weight": "model.emotional_module.0.weight",
  "emotional_module.0.bias": "model.emotional_module.0.bias",
  "emotional_module.2.weight": "model.emotional_module.2.weight",
  "emotional_module.2.bias": "model.emotional_module.2.bias",
  "feelings_layer.0.weight": "model.feelings_layer.0.weight",
  "feelings_layer.0.bias": "model.feelings_layer.0.bias",
  "extra_mlp.0.weight": "model.extra_mlp.0.weight",
  "extra_mlp.0.bias": "model.extra_mlp.0.bias",
  "extra_mlp.2.weight": "model.extra_mlp.2.weight",
  "extra_mlp.2.bias": "model.extra_mlp.2.bias",
  "classifier.weight": "model.classifier.weight",
  "classifier.bias": "model.classifier.bias"
}