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

UNet_Medical TensorFlow2 checkpoint trained with AMP

  • 1 Version
    21.11.0Selected
    02/03/2022 10:15 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