aesthetic changes

neptune-ai · May 24, 2018 · c4fed87 · c4fed87
1 parent 3048a4b
commit c4fed87
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pipelines.py b/pipelines.py
@@ -9,12 +9,8 @@
 
 
 def unet(config, train_mode):
-    if train_mode:
-        save_output = False
-        load_saved_output = False
-    else:
-        save_output = False
-        load_saved_output = False
+    save_output = False
+    load_saved_output = False
 
     loader = preprocessing(config, model_type='single', is_train=train_mode)
     unet = Step(name='unet',