diff --git a/pysegcnn/core/graphics.py b/pysegcnn/core/graphics.py index 7c9755ec48dc9a5038f4c86415fe9394a10c7d64..f8083f22fa2d4e32fc4b061de25cb0c7e5b55d82 100644 --- a/pysegcnn/core/graphics.py +++ b/pysegcnn/core/graphics.py @@ -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