Model
EfficientNet v2-S TensorFlow2 checkpoint trained on Imagenet using 1 DGX A100 (batchsize=3680=8x460)
Use the NGC CLI to download:
Copied!
1 Version
21.09.0Selected03/08/2022 12:16 AM UTC329.16 MBAccuracy: 0.8393350 EpochsBatch Size: 0GPU: A100 Copied!
21.09.0Selected
03/08/2022 12:16 AM UTC329.16 MBAccuracy: 0.8393350 EpochsBatch Size: 0GPU: A100
Copied!
architecture
| Key | Value |
|---|---|
| task | Classification |
| type | V2-S |
performance
| Key | Value |
|---|---|
| accuracy | 0.8393 |
training
| Key | Value |
|---|---|
| LR_decay | cosine |
| global_batch_size | 3680 |
| n_stages | 4 |
| EMA | 0.9999 |
| training_precision | AMP |
| scaled_LR_init | 0.1437 |
| dataset | ImageNet |
| epochs | 350 |
| warmup_epochs | 0.2 |