From 4be0892a72e9885b650bcc4c1d634fb99807a1a5 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 23 Jan 2021 13:25:14 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 1728c076..fc0c3f11 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -16,9 +16,9 @@ CACHE_DATA_IN_MEM = True
CACHE_GFS = True
DISK_CACHE = True
-PROC_BATCH_SIZE = 40
+PROC_BATCH_SIZE = 60
NumLabels = 1
-BATCH_SIZE = 512
+BATCH_SIZE = 1024
NUM_EPOCHS = 200
GLOBAL_BATCH_SIZE = BATCH_SIZE * 3
@@ -687,10 +687,10 @@ class CloudHeightNN:
labels = mini_batch[2]
pred = self.model(inputs, training=False)
t_loss = self.loss(labels, pred)
- t_loss = tf.nn.compute_average_loss(t_loss, global_batch_size=GLOBAL_BATCH_SIZE)
self.test_loss(t_loss)
self.test_accuracy(labels, pred)
+
return t_loss
@tf.function
--
GitLab