Skip to content
Snippets Groups Projects
Commit 01543403 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[script] remove the comments in the script to show stats, to display the losses

parent 6b21c0c4
Branches
No related tags found
No related merge requests found
......@@ -65,18 +65,18 @@ def main(user_input=None):
loss_filename = max(losses_files, key=os.path.getctime)
print loss_filename
#fl = bob.io.base.HDF5File(loss_filename)
#d_loss = fl.read('d_loss')
#g_loss = fl.read('g_loss')
#pyplot.title("Losses")
#pyplot.xlabel("# of iterations")
#pyplot.plot(d_loss, 'b', label="discriminator")
#pyplot.plot(g_loss, 'r', label="generator")
#pyplot.legend()
#pyplot.show()
#del fl
fl = bob.io.base.HDF5File(loss_filename)
d_loss = fl.read('d_loss')
g_loss = fl.read('g_loss')
pyplot.title("Losses")
pyplot.xlabel("# of iterations")
pyplot.plot(d_loss, 'b', label="discriminator")
pyplot.plot(g_loss, 'r', label="generator")
pyplot.legend()
pyplot.show()
del fl
# === DISCRIMINATOR ===
fdr = bob.io.base.HDF5File(logdir + '/discriminator_real_stats.hdf5')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment