{ | |
"architecture": "resnet50", | |
"as_regression": true, | |
"ema": { | |
"decay": 0.999 | |
}, | |
"img_size": [ | |
512, | |
512 | |
], | |
"lr": 3e-05, | |
"num_classes": 5, | |
"optimizer": { | |
"betas": [ | |
0.9, | |
0.999 | |
], | |
"weight_decay": 0.0005 | |
}, | |
"pretrained": true, | |
"swa": { | |
"swa_lrs": 3e-05 | |
} | |
} |