p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22746 modify for I480QV on facerecognition 

Merge pull request !22746 from 郑彬/I489QV_facerecognition
This commit is contained in:
i-robot 2021-09-02 03:19:02 +00:00 committed by Gitee
parent 6682732a33 2c6d842ff2
commit 3194d23498
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/research/cv/FaceRecognition/train.py
View File

@ -278,9 +278,9 @@ def run_train():
weight_decay=config.weight_decay)
scale_manager = DynamicLossScaleManager(init_loss_scale=config.dynamic_init_loss_scale, scale_factor=2,
scale_window=2000)
if config.device_target == "Ascend":
if config.device_target in ("Ascend", "GPU"):
model = Model(train_net, optimizer=opt, metrics=None, loss_scale_manager=scale_manager)
elif config.device_target in ("CPU", "GPU"):
elif config.device_target == "CPU":
model = Model(train_net, optimizer=opt, metrics=None, loss_scale_manager=None)
save_checkpoint_steps = config.ckpt_steps