From 53f055d2b8057a5175f7c4c2e8cdee59854828a4 Mon Sep 17 00:00:00 2001 From: zhengya01 Date: Wed, 27 Feb 2019 06:24:23 +0000 Subject: [PATCH] add ctr ce --- fluid/PaddleRec/ctr/train.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/fluid/PaddleRec/ctr/train.py b/fluid/PaddleRec/ctr/train.py index 6838df85..7ed380b6 100644 --- a/fluid/PaddleRec/ctr/train.py +++ b/fluid/PaddleRec/ctr/train.py @@ -111,16 +111,6 @@ def parse_args(): '--enable_ce', action='store_true', help='If set, run the task with continuous evaluation logs.') - parser.add_argument( - '--num_threads', - type=int, - default=1, - help='The num of threads, (default: 1)') - parser.add_argument( - '--num_cpu', - type=int, - default=1, - help='The num of cpu, (default: 1)') return parser.parse_args() @@ -266,7 +256,7 @@ def get_cards(args): cpu_num = os.environ.get('CPU_NUM', 1) return int(threads_num), int(cpu_num) else: - return args.num_threads, args.num_cpu + return 1, 1 if __name__ == '__main__': -- GitLab