Fixes the iteration count used for logging to tensorboard (FAIR-Chem#128

) * Fixes the step number used for logging to tensorboard * Trying to get the build to pass
elliottower · Oct 16, 2020 · 105175e · 105175e
1 parent 4f9bf79
commit 105175e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ocpmodels/trainers/forces_trainer.py b/ocpmodels/trainers/forces_trainer.py
@@ -422,7 +422,10 @@ def train(self):
                 # Evaluate on val set every `eval_every` iterations.
                 if eval_every != -1 and iters % eval_every == 0:
                     if self.val_loader is not None:
-                        val_metrics = self.validate(split="val", epoch=epoch)
+                        val_metrics = self.validate(
+                            split="val",
+                            epoch=epoch - 1 + (i + 1) / len(self.train_loader),
+                        )
                         if (
                             val_metrics[
                                 self.config["task"].get(