From 43c2c3bb78ac7a17d10b85aab55d2382802e8c49 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 27 Jan 2021 21:09:19 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight_mgpus.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/cloudheight_mgpus.py b/modules/deeplearning/cloudheight_mgpus.py
index fc0c3f11..05bc45c0 100644
--- a/modules/deeplearning/cloudheight_mgpus.py
+++ b/modules/deeplearning/cloudheight_mgpus.py
@@ -817,7 +817,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