snapshot...

modules/deeplearning/cloudheight_mgpus.py

@@ -817,6 +817,8 @@ class CloudHeightNN:
             print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
             ckpt_manager.save()
 
+            self.DISK_CACHE = True
+
             if self.DISK_CACHE and epoch == 0:
                 f = open(cachepath, 'wb')
                 pickle.dump(self.in_mem_data_cache, f)