From 026a2557afb57a8c8d2a1bc2de8c72b308ec6a75 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 27 Jan 2021 21:12:14 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight_mgpus.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/deeplearning/cloudheight_mgpus.py b/modules/deeplearning/cloudheight_mgpus.py
index 05bc45c0..ee1e2fb6 100644
--- a/modules/deeplearning/cloudheight_mgpus.py
+++ b/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)
--
GitLab