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
Hide whitespace changes
Inline Side-by-side
pysegcnn/core/graphics.py
+ 1
− 1
View file @ 90ccee7f
...@@ -375,7 +375,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5, ...@@ -375,7 +375,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5,
ls='--', color='grey') ls='--', color='grey')
ax1.text(esepoch - nbatches, ax1.get_ylim()[0] + 0.01, ax1.text(esepoch - nbatches, ax1.get_ylim()[0] + 0.01,
'epoch = {}, accuracy = {:.1f}%' 'epoch = {}, accuracy = {:.1f}%'
.format(int(esepoch / nbatches), esacc * 100), .format(int(esepoch / nbatches) + 1, esacc * 100),
ha='right', color='grey') ha='right', color='grey')
# create a patch (proxy artist) for every color # create a patch (proxy artist) for every color
......
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