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

UNet_Medical TensorFlow2 TF-TRT model trained with AMP

1 Version
21.11.0Selected
02/03/2022 10:17 PM UTC330.08 MBAccuracy: 00 EpochsBatch Size: 8GPU: A100
architecture
KeyValue
typebase
performance
KeyValue
eval_dice_loss0.09937536716461182
eval_dice_score0.9006246328353882
eval_total_loss0.2963957190513611
eval_ce_loss0.19702035188674927
training
KeyValue
batch_size8
training_precisionAMP
datasetEM segmentation challenge
learning_rate0.0001
iterations2000