mobilevit_s-s2-v0.1.1 / config.json
hackelle's picture
Upload mobilevit_s-s2-v0.1.1
834d50a verified
raw
history blame contribute delete
853 Bytes
{
"_fusion_activation": "nn.Tanh()",
"_fusion_method": "torch.mul",
"channels": 10,
"class_names": [
"0",
"1",
"2",
"3",
"4",
"5",
"6",
"7",
"8",
"9",
"10",
"11",
"12",
"13",
"14",
"15",
"16",
"17",
"18"
],
"classes": 19,
"custom_fusion_activation": null,
"custom_fusion_method": null,
"drop_path_rate": 0.15,
"drop_rate": 0.15,
"fusion_dropout_rate": 0.25,
"fusion_hidden": 256,
"fusion_in": 512,
"fusion_out": 512,
"hf_model_name": null,
"image_size": 120,
"load_pretrained_hf_if_available": true,
"load_pretrained_timm_if_available": false,
"max_sequence_length": 32,
"network_type": 0,
"t_dropout_rate": 0.25,
"timm_model_name": "mobilevit_s",
"use_pooler_output": true,
"v_dropout_rate": 0.25,
"visual_features_out": 512
}