| { | |
| "architectures": ["Unet"], | |
| "model_type": "unet", | |
| "backbone": "vgg19", | |
| "input_channels": 3, | |
| "num_classes": 3, | |
| "epochs": 10, | |
| "lr": 0.0002, | |
| "scheduler": "CosineAnnealingLR", | |
| "min_lr": 0.000001, | |
| "T_max": 300, | |
| "T_0": 25, | |
| "warmup_epochs": 0, | |
| "weight_decay": 0.000001, | |
| "batch_size": 64, | |
| "image_size": [224, 224], | |
| "augmentation_strategy": { | |
| "Resize": [224, 224], | |
| "HorizontalFlip": 0.5, | |
| "ShiftScaleRotate": { | |
| "shift_limit": 0.0625, | |
| "scale_limit": 0.05, | |
| "rotate_limit": 10 | |
| }, | |
| "GridDistortion": { | |
| "num_steps": 5, | |
| "distort_limit": 0.05 | |
| }, | |
| "ElasticTransform": { | |
| "alpha": 1, | |
| "sigma": 50, | |
| "alpha_affine": 50 | |
| }, | |
| "CoarseDropout": { | |
| "max_holes": 8, | |
| "max_height": 28, | |
| "max_width": 28, | |
| "min_holes": 5, | |
| "fill_value": 0 | |
| } | |
| } | |
| } | |