| { | |
| "architecture": "densenet121.tv_in1k", | |
| "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 | |
| } | |
| } |