Update train_model.py

src/models/train_model.py

@@ -32,6 +32,8 @@ def train_model():
         max_epochs=params["epochs"],
         use_gpu=params["use_gpu"],
         learning_rate=float(params["learning_rate"]),
+        num_workers=int(params["num_workers"]),
+    )
 
     model.save_model(model_dir=params['model_dir'])