change range

AirShower_WGAN/utils.py

@@ -96,7 +96,7 @@ def plot_loss(loss, log_dir = ".", name=""):
     plt.legend(loc='upper right',prop={'size':10})
     ax1.set_xlabel(r'Iterations')
     ax1.set_ylabel(r'Loss')
-    ax1.set_ylim(-1, 3)
+    ax1.set_ylim(-1.5, 1.5)
     fig.savefig(log_dir + '/%s_Loss.png' %name, dpi=120)
     plt.close('all')