Skip to content
Snippets Groups Projects
Commit 4da4e52c authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 92bddedb
Branches
No related tags found
No related merge requests found
......@@ -602,11 +602,11 @@ class SRCNN:
self.test_loss(t_loss)
self.test_accuracy(labels, pred)
@tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
# @tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
def predict(self, inputs, labels):
labels = tf.squeeze(labels)
pred = self.model([inputs], training=False)
t_loss = self.loss(labels, pred)
t_loss = self.loss(tf.squeeze(labels), pred)
self.test_labels.append(labels)
self.test_preds.append(pred.numpy())
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment