From 836249ddf5fa570be27f121e766a024413945a79 Mon Sep 17 00:00:00 2001
From: guochaorong <guochaorong@baidu.com>
Date: Thu, 9 Aug 2018 15:39:09 +0800
Subject: [PATCH] fix error in language model

---
 fluid/language_model/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fluid/language_model/train.py b/fluid/language_model/train.py
index f3e7a739..e719f8f1 100644
--- a/fluid/language_model/train.py
+++ b/fluid/language_model/train.py
@@ -145,7 +145,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()
+            gpu_num = get_cards(args.enable_ce)
             if gpu_num == 1:
                 print("kpis	imikolov_20_pass_duration	%s" %
                       (total_time / epoch_idx))
-- 
GitLab