From 90d4171e784d73f8cbced4cf675c088f5b11639d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 22 Nov 2020 20:46:51 -0600
Subject: [PATCH] minor
---
modules/deeplearning/cloudheight.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index ceede481..f6694ebd 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/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()
--
GitLab