Model

Resolution

Epochs

LR

Batch

Optimizer

Loss

HRNet

1280 × 1280

80,000

1 × 10−4

2

Adam

Cross Entropy

Resnet (both 18 & 34)

243 × 324

Frozen 15, Unfrozen 30

LR Finder

12

Adam

Cross Entropy

Resnet (both 18 & 34)

486 × 648

Frozen 15, Unfrozen 30

LR Finder

4

Adam

Cross Entropy

Resnet (both 18 & 34)

972 × 1296

Frozen 15, Unfrozen 40

LR Finder

1

Adam

Cross Entropy

Efficientnet

1280 × 1280

13

1 × 10−4

2

Adam

Cross Entropy

Detectron2

2592 × 1944

3000

25 × 10−5

8

SGD

Cross Entropy