From a9b136095e09439babc60b601a2605f8af9594c6 Mon Sep 17 00:00:00 2001 From: zhengya01 Date: Tue, 26 Feb 2019 13:48:42 +0000 Subject: [PATCH] add ctr ce --- fluid/PaddleRec/ctr/train.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/fluid/PaddleRec/ctr/train.py b/fluid/PaddleRec/ctr/train.py index bbeebafe..cf108b5d 100644 --- a/fluid/PaddleRec/ctr/train.py +++ b/fluid/PaddleRec/ctr/train.py @@ -194,7 +194,6 @@ def train_loop(args, train_program, py_reader, loss, auc_var, batch_auc_var, # only for ce if args.enable_ce: cpu_num = get_cards(args) - print("cpu_num", cpu_num) epoch_idx = args.num_passes print("kpis\teach_pass_duration_card%s\t%s" % (cpu_num, total_time / epoch_idx)) @@ -259,11 +258,8 @@ def train(): def get_cards(args): if args.enable_ce: cards = os.environ.get('NUM_THREADS', 1) - print("cards", cards) - return int(cards) else: - print("return args.num_threads") return args.num_threads -- GitLab