Sreevishnu's picture
Align label mapping with imdb dataset (#1)
70152e0
raw
history blame
980 Bytes
{
"_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,
"initializer_range": 0.1,
"initializer_std": null,
"layer_norm_eps": 1e-09,
"max_position_embeddings": 1024,
"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.19.1",
"truncate_seq": true,
"type_vocab_size": 3,
"vocab_size": 30522,
"label2id": {
"neg": 0,
"pos": 1
},
"id2label": {
"0": "neg",
"1": "pos"
}
}