NVIDIA Deep Learning Examples
UNet_Medical TensorFlow2 SavedModel (AMP)
Model
NVIDIA Deep Learning Examples
UNet_Medical TensorFlow2 SavedModel (AMP)

UNet_Medical TensorFlow2 SavedModel trained with AMP

  • 1 Version
    21.11.0Selected
    02/03/2022 10:16 PM UTC132.57 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