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
UNet_Medical TensorFlow2 checkpoint (AMP)
Model
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
UNet_Medical TensorFlow2 checkpoint (AMP)

UNet_Medical TensorFlow2 checkpoint trained with AMP

1 Version
21.11.0_ampSelected
09/22/2022 4:51 PM UTC394.99 MBAccuracy: 00 EpochsBatch Size: 8GPU: A100
architecture
KeyValue
typebase
performance
KeyValue
eval_dice_loss0.09937584400177002
eval_dice_score0.90062415599823
eval_total_loss0.29639625549316406
eval_ce_loss0.19702039659023285
training
KeyValue
batch_size8
training_precisionAMP
datasetEM segmentation challenge
learning_rate0.0001
iterations2000