OEkfxTT / config.json
Diamantis99's picture
Push model using huggingface_hub.
00c798e verified
{
"_model_class": "Unet",
"activation": null,
"aux_params": null,
"classes": 1,
"decoder_attention_type": null,
"decoder_channels": [
256,
128,
64,
32,
16
],
"decoder_interpolation": "nearest",
"decoder_use_norm": "batchnorm",
"encoder_depth": 5,
"encoder_name": "se_resnext101_32x4d",
"encoder_weights": "imagenet",
"in_channels": 3
}