diff --git a/src/ptbench/engine/trainer.py b/src/ptbench/engine/trainer.py index eedc89e8d52dde8bd1f3ef7ce190b3d06cd984f1..825ac8cd578f993e2bc874c06ad0cbdeeee0eefd 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], )