changed the error term in check_vertical_flux

taufactor/taufactor.py

@@ -149,6 +149,9 @@ def solve(self, iter_limit=5000, verbose=True, conv_crit=2*10**-2):
                 out /= self.nn
                 # check convergence using criteria
                 if self.iter % 100 == 0:
+                    # plt.imshow(self.conc[0,0,3:-2,290:-2].cpu().numpy())
+                    # print('hello')
+                    # plt.show()
                     self.converged = self.check_convergence(
                         verbose, conv_crit, start, iter_limit)
                 # efficient way of adding flux to old conc with overrelaxation
@@ -198,7 +201,7 @@ def check_vertical_flux(self, conv_crit):
         vert_flux[self.conc[:, :-2, 1:-1, 1:-1] == 0] = 0
         vert_flux[self.conc[:, 1:-1, 1:-1, 1:-1] == 0] = 0
         fl = torch.sum(vert_flux, (0, 2, 3))[1:-1]
-        err = (fl.max() - fl.min())*2/(fl.max() + fl.min())
+        err = (fl.max() - fl.min())/(fl.max())
         if fl.min() == 0:
             return 'zero_flux', torch.mean(fl), err
         if err < conv_crit or torch.isnan(err).item():