From 7d5006055f1515fa8f83f853efb18ee12a984594 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Tue, 28 Apr 2020 16:19:14 +0800 Subject: [PATCH] polish train --- tools/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/train.py b/tools/train.py index 65a0c8f90..ab7752fd1 100644 --- a/tools/train.py +++ b/tools/train.py @@ -95,7 +95,7 @@ def main(args): # 1. train with train dataset program.run(train_dataloader, exe, compiled_train_prog, train_fetchs, epoch_id, 'train') - if int(os.environ.get("PADDLE_TRAINERS_ID", 0)) == 0: + if int(os.getenv("PADDLE_TRAINER_ID", 0)) == 0: # 2. validate with validate dataset if config.validate and epoch_id % config.valid_interval == 0: top1_acc = program.run(valid_dataloader, exe,