NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
GPUNet-P0 pretrained weights (PyTorch, AMP, ImageNet)
Model
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
GPUNet-P0 pretrained weights (PyTorch, AMP, ImageNet)

GPUNet-P0 ImageNet pretrained weights

1 Version
21.12.0_ampSelected
05/10/2022 8:03 PM UTC87.44 MBAccuracy: 0450 EpochsBatch Size: 256GPU: V100
architecture
KeyValue
typeprunet_0
performance
KeyValue
val_top180.548
val_top595.026
training
KeyValue
aarand-m9-mstd0.5
batch_size256
ampTrue
decay_epochs2.4
lr0.08
crop_pct0.875
modelprunet_0
img_size224
epochs450
decay_rate0.97
model_emaTrue
model_ema_decay0.9999