| { | |
| "_model_class": "Unet", | |
| "activation": null, | |
| "aux_params": null, | |
| "classes": 1, | |
| "decoder_attention_type": null, | |
| "decoder_channels": [ | |
| 256, | |
| 128, | |
| 64, | |
| 32, | |
| 16 | |
| ], | |
| "decoder_use_batchnorm": true, | |
| "encoder_depth": 5, | |
| "encoder_name": "efficientnet-b4", | |
| "encoder_weights": "imagenet", | |
| "in_channels": 3 | |
| } |