From 4910aa7376718ac4140dc95ada9fd5691fc3929c Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 27 Jan 2021 21:11:49 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 55585f0e..c0fe2d60 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -14,7 +14,6 @@ LOG_DEVICE_PLACEMENT = False
CACHE_DATA_IN_MEM = True
CACHE_GFS = True
-DISK_CACHE = True
PROC_BATCH_SIZE = 40
PROC_BATCH_BUFFER_SIZE = 50000
@@ -242,6 +241,8 @@ class CloudHeightNN:
self.inputs.append(self.X_prof)
self.inputs.append(self.X_sfc)
+ self.DISK_CACHE = True
+
if datapath is not None:
self.DISK_CACHE = False
f = open(datapath, 'rb')
@@ -800,7 +801,7 @@ class CloudHeightNN:
print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
ckpt_manager.save()
- if DISK_CACHE and epoch == 0:
+ if self.DISK_CACHE and epoch == 0:
f = open(cachepath, 'wb')
pickle.dump(self.in_mem_data_cache, f)
f.close()
--
GitLab