RoofSense / config.json
DimitrisMantas's picture
Push model using huggingface_hub.
f34af4f verified
raw
history blame contribute delete
606 Bytes
{
"_model_class": "DeepLabV3Plus",
"activation": null,
"aux_params": null,
"block_args": {
"attn_layer": "eca"
},
"classes": 9,
"decoder_aspp_dropout": 0,
"decoder_aspp_separable": true,
"decoder_atrous_rates": [
20,
15,
6
],
"decoder_channels": 256,
"drop_path_rate": 0.43231510659674255,
"drop_rate": 0.030244232449387346,
"encoder_depth": 5,
"encoder_name": "tu-resnet18d",
"encoder_output_stride": 16,
"encoder_weights": "imagenet",
"global_pool": "avgmax",
"in_channels": 10,
"upsampling": 4,
"zero_init_last": true
}