提交 4d579648 编写于 作者: Z zhengya01

add ctr ce

上级 53f055d2
...@@ -251,12 +251,9 @@ def train(): ...@@ -251,12 +251,9 @@ def train():
def get_cards(args): def get_cards(args):
if args.enable_ce: threads_num = os.environ.get('NUM_THREADS', 1)
threads_num = os.environ.get('NUM_THREADS', 1) cpu_num = os.environ.get('CPU_NUM', 1)
cpu_num = os.environ.get('CPU_NUM', 1) return int(threads_num), int(cpu_num)
return int(threads_num), int(cpu_num)
else:
return 1, 1
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册