spandan21's picture
Upload config.json
c2d8c4a verified
{
"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
}
}
}