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

Fixed a bug: correct number of trained epochs is now displayed.

parent 2ff46ca3
No related branches found
No related tags found
No related merge requests found
......@@ -375,7 +375,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5,
ls='--', color='grey')
ax1.text(esepoch - nbatches, ax1.get_ylim()[0] + 0.01,
'epoch = {}, accuracy = {:.1f}%'
.format(int(esepoch / nbatches), esacc * 100),
.format(int(esepoch / nbatches) + 1, esacc * 100),
ha='right', color='grey')
# create a patch (proxy artist) for every color
......
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