mito-aff-unetiso-setup-53 / torch_source.json
zouinkhim's picture
Add PyTorch original: model_state_dict.pt + model_def.py + torch_source.json
5b16080 verified
Raw History Blame Contribute Delete
205 Bytes
{
"module": "model_def.py",
"class": "Unet3dIso",
"kwargs": {
"num_features": 32,
"feature_factor": 2,
"out_channels": 3
},
"state_dict": "model_state_dict.pt",
"torch_version": "2.9.1+cu128"
}