6afwO9u / config.json
Diamantis99's picture
Push model using huggingface_hub.
7b50cba verified
{
"_model_class": "PSPNet",
"activation": null,
"aux_params": null,
"classes": 1,
"decoder_use_norm": "batchnorm",
"encoder_depth": 3,
"encoder_name": "inceptionresnetv2",
"encoder_weights": "imagenet",
"in_channels": 3,
"psp_dropout": 0.2,
"psp_out_channels": 512,
"upsampling": 8
}