File size: 318 Bytes
3628aa4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
{
  "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
  }
}