Update base.py

Author: ZebinYang

exnn/base.py

@@ -572,8 +572,8 @@ def visualize_new(self, cols_per_row=3, subnet_num=10**5, dummy_subnet_num=10**5
             if save_png:
                 if not os.path.exists(folder):
                     os.makedirs(folder)
-                f.savefig("%s.png" % save_path, bbox_inches='tight', dpi=100)
+                fig.savefig("%s.png" % save_path, bbox_inches='tight', dpi=100)
             if save_eps:
                 if not os.path.exists(folder):
                     os.makedirs(folder)
-                f.savefig("%s.eps" % save_path, bbox_inches='tight', dpi=100)
+                fig.savefig("%s.eps" % save_path, bbox_inches='tight', dpi=100)