Fix bug in loggin buffer reset (ecmwf#651)

clessig · clessig · web-flow · commit 45390fb51a11 · 2025-07-31T14:30:45.000+02:00
Co-authored-by: clessig &lt;christian.lessig@ecwmf.int&gt;
diff --git a/src/weathergen/train/trainer.py b/src/weathergen/train/trainer.py
@@ -694,7 +694,7 @@ def _log(self, stage: Stage):
                     self.perf_mem,
                 )
 
-            self.loss_unweighted_hist, self.loss_model_hist, self.stdev_unweighted_hist = [], [], []
+        self.loss_unweighted_hist, self.loss_model_hist, self.stdev_unweighted_hist = [], [], []
 
     def _log_terminal(self, bidx: int, epoch: int, stage: Stage):
         if bidx % self.print_freq == 0 and bidx > 0 or stage == VAL:

Original file line number	Diff line number	Diff line change
`@@ -694,7 +694,7 @@ def _log(self, stage: Stage):`
`694`	`694`	`self.perf_mem,`
`695`	`695`	`)`
`696`	`696`
`697`		`- self.loss_unweighted_hist, self.loss_model_hist, self.stdev_unweighted_hist = [], [], []`
	`697`	`+ self.loss_unweighted_hist, self.loss_model_hist, self.stdev_unweighted_hist = [], [], []`
`698`	`698`
`699`	`699`	`def _log_terminal(self, bidx: int, epoch: int, stage: Stage):`
`700`	`700`	`if bidx % self.print_freq == 0 and bidx > 0 or stage == VAL:`