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

!3944 modify glore_res scripts

Merge pull request !3944 from anzhengqi/modify-glore-scripts
parents aa1fa7e2 017254b4
No related branches found
No related tags found
No related merge requests found
...@@ -46,14 +46,14 @@ from src.save_callback import SaveCallback ...@@ -46,14 +46,14 @@ from src.save_callback import SaveCallback
if config.isModelArts: if config.isModelArts:
import moxing as mox import moxing as mox
if config.net == 'resnet200' or config.net == 'resnet101' or config.net == 'resnet50': if config.net == 'resnet200' or config.net == 'resnet101':
if config.device_target == "GPU": if config.device_target == "GPU":
config.cast_fp16 = False config.cast_fp16 = False
random.seed(1) random.seed(1)
np.random.seed(1) np.random.seed(1)
de.config.set_seed(1) de.config.set_seed(1)
if config.net == 'resnet200' or config.net == 'resnet101': if config.net == 'resnet200' or config.net == 'resnet101' or config.net == 'resnet50':
set_seed(1) set_seed(1)
if __name__ == '__main__': if __name__ == '__main__':
......
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