From f97adda311528915c9c9e94cd1b768494331c43f Mon Sep 17 00:00:00 2001 From: liuhongyu Date: Fri, 9 Nov 2018 11:48:53 +0800 Subject: [PATCH] fix gru lm --- fluid/PaddleNLP/language_model/gru/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/PaddleNLP/language_model/gru/train.py b/fluid/PaddleNLP/language_model/gru/train.py index a999d37e..84b7e15a 100644 --- a/fluid/PaddleNLP/language_model/gru/train.py +++ b/fluid/PaddleNLP/language_model/gru/train.py @@ -148,7 +148,7 @@ def train(train_reader, if pass_idx == pass_num - 1 and args.enable_ce: #Note: The following logs are special for CE monitoring. #Other situations do not need to care about these logs. - gpu_num = get_cards(args.enable_ce) + gpu_num = get_cards(args) if gpu_num == 1: print("kpis imikolov_20_pass_duration %s" % (total_time / epoch_idx)) -- GitLab