omerfguzel's picture
End of training
6ff94ff verified
{
"_name_or_path": "funnel-transformer/small",
"activation_dropout": 0.0,
"architectures": [
"FunnelForSequenceClassification"
],
"attention_dropout": 0.1,
"attention_type": "relative_shift",
"block_repeats": [
1,
1,
1
],
"block_sizes": [
4,
4,
4
],
"d_head": 64,
"d_inner": 3072,
"d_model": 768,
"hidden_act": "gelu_new",
"hidden_dropout": 0.1,
"id2label": {
"0": "sadness",
"1": "joy",
"2": "love",
"3": "anger",
"4": "fear",
"5": "surprise"
},
"initializer_range": 0.1,
"initializer_std": null,
"label2id": {
"anger": 3,
"fear": 4,
"joy": 1,
"love": 2,
"sadness": 0,
"surprise": 5
},
"layer_norm_eps": 1e-09,
"max_position_embeddings": 512,
"model_type": "funnel",
"n_head": 12,
"num_decoder_layers": 2,
"pool_q_only": true,
"pooling_type": "mean",
"problem_type": "single_label_classification",
"rel_attn_type": "factorized",
"separate_cls": true,
"torch_dtype": "float32",
"transformers_version": "4.37.2",
"truncate_seq": true,
"type_vocab_size": 3,
"vocab_size": 30522
}