fix a bug in language model (#4120)

PaddleNLP/language_model/train.py

@@ -125,7 +125,6 @@ def main():
             res_vars = lm_model.lm_model(
                 config.hidden_size,
                 config.vocab_size,
-                config.batch_size,
                 num_layers=config.num_layers,
                 num_steps=config.num_steps,
                 init_scale=config.init_scale,
@@ -160,7 +159,6 @@ def main():
             lm_model.lm_model(
                 config.hidden_size,
                 config.vocab_size,
-                config.batch_size,
                 num_layers=config.num_layers,
                 num_steps=config.num_steps,
                 init_scale=config.init_scale,
@@ -319,7 +317,7 @@ def main():
                 print(
                     "-- Epoch:[%d]; Batch:[%d]; Time: %.5f s; ppl: %.5f, lr: %.5f"
                     % (epoch_id, batch_id, batch_time, ppl[0], lr[0]))
-            
+
             # profiler tools for benchmark
             if args.profile and batch_id == log_interval:
                 profiler.reset_profiler()

PaddleNLP/models/language_model/lm_model.py

@@ -26,7 +26,6 @@ from paddle.fluid.contrib.layers import basic_lstm
 
 def lm_model(hidden_size,
              vocab_size,
-             batch_size,
              num_layers=2,
              num_steps=20,
              init_scale=0.1,
@@ -253,7 +252,6 @@ def lm_model(hidden_size,
 
         return real_res, last_hidden, last_cell
 
-    batch_size_each = batch_size // fluid.core.get_cuda_device_count()
     x = fluid.data(name="x", shape=[None, num_steps, 1], dtype='int64')
     y = fluid.data(name="y", shape=[None, 1], dtype='int64')