mobilesam / config.json
bhllx's picture
Update config.json
9b7ada1 verified
raw
history blame contribute delete
717 Bytes
{
"architectures": [
"SamModel"
],
"initializer_range": 0.02,
"mask_decoder_config": {
"model_type": ""
},
"model_type": "sam",
"prompt_encoder_config": {
"model_type": ""
},
"torch_dtype": "float32",
"transformers_version": "4.37.0.dev0",
"vision_config": {
"model_type": "",
"img_size": 1024,
"in_chans": 3,
"num_classes": 1000,
"embed_dims": [64, 128, 160, 320],
"depths": [2, 2, 6, 2],
"num_heads": [2, 4, 5, 10],
"window_sizes": [7, 7, 14, 7],
"mlp_ratio": 4.0,
"drop_rate": 0.0,
"drop_path_rate": 0.0,
"use_checkpoint": false,
"mbconv_expand_ratio": 4.0,
"local_conv_size": 3,
"layer_lr_decay": 0.8
}
}