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

!16235 make device_id can be getted from environment and modify required version for package pycocotools 

From: @zhouneng2
Reviewed-by: @oacjiewen,@liangchenghui
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2021-05-12 09:34:44 +08:00 committed by Gitee
parent 507bf8c5c8 7ca9108982
commit fdd0c73cd5
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model_zoo/official/cv/xception/train.py
View File

@ -64,9 +64,8 @@ if __name__ == '__main__':
else: else:
rank = 0 rank = 0
group_size = 1 group_size = 1
context.set_context(device_id=0) device_id = int(os.getenv('DEVICE_ID', '0'))
# if os.getenv('DEVICE_ID', "not_set").isdigit(): context.set_context(device_id=device_id)
# context.set_context(device_id=int(os.getenv('DEVICE_ID')))
context.set_context(mode=context.GRAPH_MODE, device_target=args_opt.device_target, save_graphs=False) context.set_context(mode=context.GRAPH_MODE, device_target=args_opt.device_target, save_graphs=False)
# define network # define network

2
requirements.txt
View File

@ -15,7 +15,7 @@ sklearn >= 0.0 # for st test
pandas >= 1.0.2 # for ut test pandas >= 1.0.2 # for ut test
astunparse >= 1.6.3 astunparse >= 1.6.3
packaging >= 20.0 packaging >= 20.0
pycocotools >= 2.0.0 # for st test pycocotools >= 2.0.2 # for st test
tables >= 3.6.1 # for st test tables >= 3.6.1 # for st test
psutil >= 5.6.1 psutil >= 5.6.1
subword-nmt>=0.3.7 # for st test subword-nmt>=0.3.7 # for st test