From 835a00771c4563f8986e16b9cfd127193d169f2b Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Thu, 21 Jan 2021 12:20:01 -0600 Subject: [PATCH] snapshot... --- modules/deeplearning/cloudheight.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py index c77ac2f5..8316f11c 100644 --- a/modules/deeplearning/cloudheight.py +++ b/modules/deeplearning/cloudheight.py @@ -691,6 +691,7 @@ class CloudHeightNN: self.test_loss(t_loss) self.test_accuracy(labels, pred) + return t_loss @tf.function def distributed_train_step(self, dataset_inputs): -- GitLab