| { |
| "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 |
| } |
| } |
| } |
|
|
