Use the NGC CLI to download:
Copied!
Model
EfficientNet-B4 QAT checkpoint based on best FP32 checkpoint
Use the NGC CLI to download:
Copied!
1 Version
21.03.0_fp32Selected09/22/2022 6:02 PM UTC74.97 MBAccuracy: 0.82542 EpochsBatch Size: 32GPU: V100Copied!
21.03.0_fp32Selected
09/22/2022 6:02 PM UTC74.97 MBAccuracy: 0.82542 EpochsBatch Size: 32GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| task | pretraining |
| type | base |
performance
| Key | Value |
|---|---|
| accuracy | 0.8254 |
| training_loss | 0.69253 |
training
| Key | Value |
|---|---|
| batch_size | 32 |
| lr-schedule | linear |
| optimizer | rmsprop |
| LR | 4.09e-06 |
| training_precision | FP32 |
| dataset | ImageNet |
| epochs | 2 |
| weight decay | 4.5e-05 |
| rmsprop alpha | 0.853 |
| rmsprop epsilon | 0.00422 |
| momentum | 0.9 |