Try not to round num_updates (facebookresearch#1836)

ErikEkstedt · Jul 10, 2019 · 742210b · 742210b
1 parent f57f5ac
commit 742210b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/parlai/scripts/train_model.py b/parlai/scripts/train_model.py
@@ -585,6 +585,8 @@ def _average_dicts(self, all_versions):
             else:
                 # all other cases, take the mean across the workers
                 finalized[k] = np.mean(values)
+                if all(isinstance(v, int) for v in values):
+                    finalized[k] = int(finalized[k])
         return finalized
 
     def _cleanup_inaccurate_metrics(self, metrics):