Skip to content
Snippets Groups Projects
Commit b0655852 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[engine.trainer] Log at every epoch (closes #28)

parent 8ad8565c
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
Pipeline #77067 passed
......@@ -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],
)
......
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