Skip to content
Snippets Groups Projects
Unverified Commit 5da9e423 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3568 Enable sens dtype correct for deeptext network in PyNative mode

Merge pull request !3568 from JoyLvliang/enable_dtype_correct_for_pynative_deeptext_network
parents 560ee514 ba3c19ed
No related branches found
No related tags found
No related merge requests found
......@@ -168,6 +168,8 @@ def run_train():
net.to_float(mstype.float16)
loss = LossNet()
if device_type == "Ascend":
loss.to_float(mstype.float32)
lr = Tensor(dynamic_lr(config, rank_size=device_num), mstype.float32)
opt = Momentum(params=net.trainable_params(), learning_rate=lr, momentum=config.momentum,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment