Commit 3fe8e126 authored by Artem Pulkin's avatar Artem Pulkin
Browse files

workflows: print memory footprint and gc.collect before epoch

parent 1b7cb35b
Pipeline #86028 passed with stages
in 27 minutes and 19 seconds
......@@ -1470,8 +1470,9 @@ class FitWorkflow(Workflow):
self.log_static_memory_report()
for epoch in range(n_epochs):
self.log.info(f"Epoch {epoch:d}")
self.epoch(cleanup=cleanup_optimizer, epoch_size=epoch_size)
gc.collect()
self.log_runtime_memory_report()
self.epoch(cleanup=cleanup_optimizer, epoch_size=epoch_size)
self.update_loss(**update_kwargs)
if save:
self.build_potentials()
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment