Skip to content
Snippets Groups Projects
Commit 463be8e9 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Added logging info when plotting model training statistics.

parent 30f99e1f
No related branches found
No related tags found
No related merge requests found
......@@ -479,7 +479,9 @@ def plot_loss(state_file, figsize=(10, 10), step=5,
"""
# load the model state
state_file = pathlib.Path(state_file)
model_state = torch.load(state_file)
LOGGER.info('Plot model statistics: {}'.format(state_file.stem))
# get all non-zero elements, i.e. get number of epochs trained before
# early stop
......
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