indent saving part

MDIL-SNU · Sep 4, 2018 · 231c265 · 231c265
1 parent 818857d
commit 231c265
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/simple_nn/models/neural_network.py b/simple_nn/models/neural_network.py
@@ -628,16 +628,16 @@ def train(self, user_optimizer=None, user_atomic_weights_function=None):
 
                     # Temp saving
                     #if (epoch+1) % self.inputs['save_interval'] == 0:
-                    if save_stack > self.inputs['save_interval'] and prev_eloss > eloss and \
-                       ((prev_floss > floss) or floss == 0.):
-                        temp_time = timeit.default_timer()
-                        self._save(sess, saver)
-                        prev_eloss = eloss
-                        prev_floss = floss
-                        save_stack = 1
-                        save_time = timeit.default_timer() - temp_time
-                    else:
-                        save_time = 0
+                        if save_stack > self.inputs['save_interval'] and prev_eloss > eloss and \
+                           ((prev_floss > floss) or floss == 0.):
+                            temp_time = timeit.default_timer()
+                            self._save(sess, saver)
+                            prev_eloss = eloss
+                            prev_floss = floss
+                            save_stack = 1
+                            save_time = timeit.default_timer() - temp_time
+                        else:
+                            save_time = 0
 
                 #self._save(sess, saver)