| { | |
| "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", | |
| "wrapper": "student_2d.py:build_model (3D layers -> 2D on the single z-slice; params '<conv>.c.*')" | |
| } |