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
SE(3)-Transformer checkpoint (PyTorch, AMP, QM9 homo task)
Model
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
SE(3)-Transformer checkpoint (PyTorch, AMP, QM9 homo task)

SE(3)-Transformer checkpoint trained on QM9 homo task

1 Version
21.07.0_amp_bs-240_epochs-300Selected
09/22/2022 5:26 PM UTC106.36 MBAccuracy: 0.03424300 EpochsBatch Size: 240GPU: A100
architecture
KeyValue
tasktraining
typebase
performance
KeyValue
accuracy0.03424
training_loss0.03307
training
KeyValue
batch_size240
LR_phase20.0001
LR_phase10.004
training_precisionAMP
bs_phase2240
bs_phase1240
datasetQM9
epochs300
iterations300