nuc-resunet-setup-61 / torch_source.json
zouinkhim's picture
Add PyTorch original: model_state_dict.pt + model_def.py + torch_source.json
675c828 verified
Raw History Blame Contribute Delete
228 Bytes
{
"module": "model_def.py",
"class": "ResUNet3DValid",
"kwargs": {
"out_channels": 1,
"base_channels": 40,
"factor": 2,
"blocks_per_level": 1
},
"state_dict": "model_state_dict.pt",
"torch_version": "2.9.1+cu128"
}