diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index c74626644c6cbfedbffe9d5a4c52304ab2c152a5..f57c5599855fe76e8f1bf2e90481b2e12ae26804 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -19,7 +19,7 @@ DISK_CACHE = True
 PROC_BATCH_SIZE = 40
 NumLabels = 1
 BATCH_SIZE = 384
-NUM_EPOCHS = 120
+NUM_EPOCHS = 200
 
 
 TRACK_MOVING_AVERAGE = False
@@ -241,7 +241,8 @@ class CloudHeightNN:
         self.inputs.append(self.X_prof)
         self.inputs.append(self.X_sfc)
 
-        if (not DISK_CACHE) and (datapath is not None):
+        if datapath is not None:
+            self.DISK_CACHE = False
             f = open(datapath, 'rb')
             self.in_mem_data_cache = pickle.load(f)
             f.close()