Use the NGC CLI to download:
Copied!
Model
EfficientNet v2-S TensorFlow2 savedmodel, trained on Imagenet using 1 DGX1 V100 (batchsize=1024=8x128)
Use the NGC CLI to download:
Copied!
1 Version
21.09.1_ampSelected09/22/2022 6:02 PM UTC329.16 MBAccuracy: 0.8403350 EpochsBatch Size: 0GPU: V100Copied!
21.09.1_ampSelected
09/22/2022 6:02 PM UTC329.16 MBAccuracy: 0.8403350 EpochsBatch Size: 0GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| task | Classification |
| type | V2-S |
performance
| Key | Value |
|---|---|
| accuracy | 0.8403 |
training
| Key | Value |
|---|---|
| LR_decay | cosine |
| global_batch_size | 1024 |
| n_stages | 4 |
| EMA | 0.9999 |
| training_precision | AMP |
| scaled_LR_init | 0.04 |
| dataset | ImageNet |
| epochs | 350 |
| warmup_epochs | 5 |