diff --git a/tutorials/source_en/advanced_use/mixed_precision.md b/tutorials/source_en/advanced_use/mixed_precision.md index de7318de531f1ca3fcfdf34885358714900d960b..d7b4fb45329c0558315f63f6617846baf9499487 100644 --- a/tutorials/source_en/advanced_use/mixed_precision.md +++ b/tutorials/source_en/advanced_use/mixed_precision.md @@ -86,7 +86,7 @@ net_with_loss = WithLossCell(net, loss) train_network = amp.build_train_network(net_with_loss, optimizer, level="O2") # Run training -output = train_network(inputs, label, scaling_sens) +output = train_network(predict, label, scaling_sens) ``` @@ -143,5 +143,5 @@ net_with_loss = WithLossCell(net, loss) train_network = TrainOneStepWithLossScaleCell(net_with_loss, optimizer) # Run training -output = train_network(inputs, label, scaling_sens) +output = train_network(predict, label, scaling_sens) ```