add ctr ce

4d579648 · zhengya01 · 53f055d2 · 4d579648
隐藏空白更改
内联并排

Showing with 3 addition and 6 deletion

fluid/PaddleRec/ctr/train.py fluid/PaddleRec/ctr/train.py +3 -6

未找到文件。
--- a/fluid/PaddleRec/ctr/train.py
+++ b/fluid/PaddleRec/ctr/train.py
@@ -251,12 +251,9 @@ def train():


 def get_cards(args):
-    if args.enable_ce:
-        threads_num = os.environ.get('NUM_THREADS', 1)
-        cpu_num = os.environ.get('CPU_NUM', 1)
-        return int(threads_num), int(cpu_num)
-    else:
-        return 1, 1
+    threads_num = os.environ.get('NUM_THREADS', 1)
+    cpu_num = os.environ.get('CPU_NUM', 1)
+    return int(threads_num), int(cpu_num)


 if __name__ == '__main__':