Clear previous update on epoch

1 files changed, 1 insertions, 0 deletions

diff --git a/cgan.py b/cgan.py
index 716e5ae..3c89121 100644
--- a/cgan.py
+++ b/cgan.py
@@ -156,6 +156,7 @@ class CGAN():
             # Plot the progress
             #print ("%d [D loss: %f, acc.: %.2f%%] [G loss: %f]" % (epoch, d_loss[0], 100*d_loss[1], g_loss))
             if epoch % 500 == 0:
+              clear_output() 
               print(epoch)
             loss[0][epoch] = d_loss[0]
             loss[1][epoch] = g_loss