diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py index b1b373b6..d0f462ce 100644 --- a/pvnet/models/base_model.py +++ b/pvnet/models/base_model.py @@ -491,9 +491,9 @@ def validation_step(self, batch: dict, batch_idx): ) self.log_dict( - logged_losses, - on_step=False, - on_epoch=True, + logged_losses, + on_step=False, + on_epoch=True, ) accum_batch_num = batch_idx // self.trainer.accumulate_grad_batches