Skip to content
Snippets Groups Projects
Commit a9a858f0 authored by Sam Eadie's avatar Sam Eadie
Browse files

Merge branch 'logger_bug' into 'VTM-11.0_nnvc'

inverse logic logger

See merge request jvet-ahg-nnvc/VVCSoftware_VTM!112
parents 68eed4a2 f3b9b200
No related branches found
No related tags found
No related merge requests found
......@@ -144,7 +144,7 @@ class PrintLogger(BaseLogger):
def on_train_iter_end(
self, epoch: int, iteration: int, train_metrics: Dict[str, Any]
) -> None:
if self.log_train_interval > 0 and (iteration + 1) % self.log_train_interval:
if self.log_train_interval > 0 and (iteration + 1) % self.log_train_interval == 0:
print(
f"Epoch {epoch}, iteration {iteration}: {self.format_metrics(train_metrics)}",
file=self.out_file,
......@@ -280,7 +280,7 @@ class TensorboardLogger(BaseLogger):
def on_train_iter_end(
self, epoch: int, iteration: int, train_metrics: Dict[str, Any]
) -> None:
if self.log_train_interval > 0 and (iteration + 1) % self.log_train_interval:
if self.log_train_interval > 0 and (iteration + 1) % self.log_train_interval == 0:
self.global_iteration += self.log_train_interval
for metric, value in train_metrics.items():
self.writer.add_scalar(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment