ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
GPUNet-D1 pretrained weights (PyTorch, AMP, ImageNet)
Model
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
GPUNet-D1 pretrained weights (PyTorch, AMP, ImageNet)

GPUNet-D1 weights pretrained on ImageNet

1 Version
21.12.0_ampSelected
05/10/2022 7:54 PM UTC162.5 MBAccuracy: 0450 EpochsBatch Size: 192GPU: V100
architecture
KeyValue
typegpunet_d1
performance
KeyValue
val_top182.44
val_top596.252
training
KeyValue
aarand-m9-mstd0.5
drop0.3
drop_connect0.2
batch_size192
ampTrue
decay_epochs2.4
crop_pct1.0
img_size456
enable_distillTrue
epochs450
decay_rate0.97
data_dir/data/imagenet