diff --git a/fluid/PaddleNLP/language_model/gru/train.py b/fluid/PaddleNLP/language_model/gru/train.py
index a999d37e4142044b5ae9340d2db514cb1183c4e1..84b7e15abfaaa5d15de5a6c72db398eadcca209d 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))