minor

modules/deeplearning/cloudheight.py

@@ -797,15 +797,15 @@ class CloudHeightNN:
             print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
             ckpt_manager.save()
 
+            if DISK_CACHE and epoch == 0:
+                f = open(cachepath, 'wb')
+                pickle.dump(self.in_mem_data_cache, f)
+                f.close()
+
         print('total time: ', total_time)
         self.writer_train.close()
         self.writer_valid.close()
 
-        if DISK_CACHE:
-            f = open(cachepath, 'wb')
-            pickle.dump(self.in_mem_data_cache, f)
-            f.close()
-
     def build_model(self):
         flat = self.build_cnn()
         flat_1d = self.build_1d_cnn()