| { | |
| "module": "model_def.py", | |
| "class": "ResUNet3DValid", | |
| "kwargs": { | |
| "out_channels": 3, | |
| "base_channels": 40, | |
| "factor": 2, | |
| "blocks_per_level": 1 | |
| }, | |
| "state_dict": "model_state_dict.pt", | |
| "torch_version": "2.9.1+cu128", | |
| "wrapper": "student_2d.py:build_model (3D layers -> 2D on the single z-slice; params '<conv>.c.*')" | |
| } |