Spaces:

CVPR
/

Demo-Balanced-MSE

Runtime error

jiawei011 commited on Apr 26, 2022

Commit

6e9ac12

1 Parent(s): 7d67b25

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -227,7 +227,7 @@ def visualize(training_df, df, save_path, fast=False, epoch=None):
     plt.gca().axes.set_xlabel(r'$x$')
     plt.gca().axes.set_ylabel(r'$y$')
-    plt.savefig(save_path, bbox_inches='tight', dpi=200)
     plt.close()

     plt.gca().axes.set_xlabel(r'$x$')
     plt.gca().axes.set_ylabel(r'$y$')
+    plt.savefig(save_path, bbox_inches='tight', dpi=50 if fast else 100)
     plt.close()