Update train_model.py

src/models/train_model.py

@@ -20,6 +20,7 @@ def train_model():
 
     model.train(train_df=train_df, eval_df=eval_df,
                 batch_size=params['batch_size'], max_epochs=params['max_epoch'],
+                use_gpu=params['use_gpu'], learning_rate=params['learning_rate'],
 
 
 if __name__ == '__main__':