Syed-Hasan-8503's picture
Upload TransformerWithHeads
06e8851 verified
raw
history blame
378 Bytes
{"emotion_head_3": {"name": "emotion_head_3", "in_size": 4096, "num_outputs": 6, "layer_hook": -3, "hidden_size": 0, "num_layers": 1, "output_activation": "linear", "target": "emotion_head_3", "is_causal_lm": false, "pred_for_sequence": true, "is_regression": false, "output_bias": false, "loss_fct": "cross_entropy", "trainable": true, "loss_weight": 1.0, "ignore_pads": true}}