soumickmj's picture
Upload UNetFTSR
301db00 verified
raw
history blame contribute delete
389 Bytes
{
"architectures": [
"UNetFTSR"
],
"auto_map": {
"AutoConfig": "UNetFTSRConfig.UNetFTSRConfig",
"AutoModel": "UNetFTSR.UNetFTSR"
},
"batch_norm": false,
"depth": 3,
"dropout": false,
"in_channels": 1,
"model_type": "UNet",
"n_classes": 1,
"padding": true,
"torch_dtype": "float32",
"transformers_version": "4.44.2",
"up_mode": "upconv",
"wf": 6
}