Skip to content

Commit

Permalink
fix pytest warning: The number of training batches (1) is smaller tha…
Browse files Browse the repository at this point in the history
…n the logging interval Trainer(log_every_n_steps=50). Set a lower value for log_every_n_steps if you want to see logs for the training epoch.
  • Loading branch information
BerndDoser committed Sep 2, 2024
1 parent 15287cc commit fb78432
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/test_rotational_autoencoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ def test_training(shape_path):
overfit_batches=2,
enable_checkpointing=False,
accelerator="cpu",
log_every_n_steps=1,
)
trainer.fit(model, datamodule=datamodule)

Expand Down

0 comments on commit fb78432

Please sign in to comment.