NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
EfficientNet-B0 QAT checkpoint (PyTorch FP32)
Model
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
EfficientNet-B0 QAT checkpoint (PyTorch FP32)

EfficientNet-B0 QAT checkpoint based on best FP32 checkpoint

1 Version
21.03.0_fp32Selected
09/22/2022 6:00 PM UTC20.64 MBAccuracy: 0.771210 EpochsBatch Size: 128GPU: V100
architecture
KeyValue
taskpretraining
typebase
performance
KeyValue
accuracy0.7712
training_loss1.00662
training
KeyValue
batch_size128
lr-schedulecosine
optimizerSGD
LR0.0125
training_precisionFP32
datasetImageNet
epochs10
weight decay4.5e-05
momentum0.89