Memory usage to Tensorboard.

eduamf · eduamf · commit 2bd6b19bc21d · 2019-07-02T20:58:26.000-03:00
diff --git a/onmt/trainer.py b/onmt/trainer.py
@@ -429,11 +429,11 @@ def _maybe_report_training(self, step, num_steps, learning_rate,
         """
         Simple function to report training stats (if report_manager is set)
         see `onmt.utils.ReportManagerBase.report_training` for doc
+        included total of gpu
         """
         if self.report_manager is not None:
             return self.report_manager.report_training(
-                step, num_steps, learning_rate, report_stats,
-                multigpu=self.n_gpu > 1)
+                step, num_steps, learning_rate, report_stats, self.n_gpu)
 
     def _report_step(self, learning_rate, step, train_stats=None,
                      valid_stats=None):
