From a37b03d9e81c08ed7b0e449ae6762093e2b65536 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 16 Apr 2021 14:12:26 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index 6c3ee902..5e67a711 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -426,7 +426,7 @@ class IcingIntensityNN:
@tf.function
def train_step(self, mini_batch):
- inputs = [mini_batch[0], mini_batch[1], mini_batch[2]]
+ inputs = [mini_batch[0], mini_batch[1]]
labels = mini_batch[2]
with tf.GradientTape() as tape:
pred = self.model(inputs, training=True)
--
GitLab