soumickmj's picture
Upload UNet3D
864135f verified
raw
history blame contribute delete
281 Bytes
{
"architectures": [
"UNet3D"
],
"auto_map": {
"AutoConfig": "UNetConfigs.UNet3DConfig",
"AutoModel": "UNets.UNet3D"
},
"in_ch": 1,
"init_features": 64,
"model_type": "UNet",
"out_ch": 1,
"torch_dtype": "float32",
"transformers_version": "4.44.2"
}