From 03a03ab13d12e6c10be9abb04da4c9fa28c1f5b4 Mon Sep 17 00:00:00 2001 From: Xiaoda Zhang Date: Wed, 29 Jul 2020 14:24:07 +0800 Subject: [PATCH] changing mix-precision level --- tutorials/source_en/advanced_use/mixed_precision.md | 2 +- tutorials/source_zh_cn/advanced_use/mixed_precision.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/source_en/advanced_use/mixed_precision.md b/tutorials/source_en/advanced_use/mixed_precision.md index b88cbd8ec2..c4b942be89 100644 --- a/tutorials/source_en/advanced_use/mixed_precision.md +++ b/tutorials/source_en/advanced_use/mixed_precision.md @@ -84,7 +84,7 @@ label = Tensor(np.zeros([64, 128]).astype(np.float32)) # Define Loss and Optimizer loss = nn.SoftmaxCrossEntropyWithLogits() optimizer = Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9) -train_network = amp.build_train_network(net, optimizer, loss, level="O2", loss_scale_manager=None) +train_network = amp.build_train_network(net, optimizer, loss, level="O3", loss_scale_manager=None) # Run training output = train_network(predict, label) diff --git a/tutorials/source_zh_cn/advanced_use/mixed_precision.md b/tutorials/source_zh_cn/advanced_use/mixed_precision.md index 4cacb8c7a4..dbea7ee0a2 100644 --- a/tutorials/source_zh_cn/advanced_use/mixed_precision.md +++ b/tutorials/source_zh_cn/advanced_use/mixed_precision.md @@ -83,7 +83,7 @@ label = Tensor(np.zeros([64, 128]).astype(np.float32)) # Define Loss and Optimizer loss = nn.SoftmaxCrossEntropyWithLogits() optimizer = Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9) -train_network = amp.build_train_network(net, optimizer, loss, level="O2", loss_scale_manager=None) +train_network = amp.build_train_network(net, optimizer, loss, level="O3", loss_scale_manager=None) # Run training output = train_network(predict, label) -- Gitee