From b0655852df4ddd2266d1b042bcd524a0467d0119 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Mon, 14 Aug 2023 20:57:51 +0200
Subject: [PATCH] [engine.trainer] Log at every epoch (closes #28)
---
src/ptbench/engine/trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/ptbench/engine/trainer.py b/src/ptbench/engine/trainer.py
index eedc89e8..825ac8cd 100644
--- a/src/ptbench/engine/trainer.py
+++ b/src/ptbench/engine/trainer.py
@@ -205,6 +205,7 @@ def run(
accumulate_grad_batches=batch_chunk_count,
logger=tensorboard_logger,
check_val_every_n_epoch=1,
+ log_every_n_steps=len(datamodule.train_dataloader()),
callbacks=[LoggingCallback(resource_monitor), checkpoint_callback],
)
--
GitLab