| { | |
| "architectures": [ | |
| "UNetForSegmentation" | |
| ], | |
| "batch_size": 8, | |
| "downsample_factor": 1.0, | |
| "dtype": "float32", | |
| "img_size": 128, | |
| "k": 5, | |
| "model_arch": "unet", | |
| "model_type": "segmentation", | |
| "n_filts": 128, | |
| "norm": true, | |
| "num_channels": 3, | |
| "num_classes": 4, | |
| "t": 3, | |
| "transformers_version": "5.3.0" | |
| } | |