Skip to content
Snippets Groups Projects
Commit 1c7408f3 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 4da4e52c
Branches
No related tags found
No related merge requests found
......@@ -603,6 +603,7 @@ class SRCNN:
self.test_accuracy(labels, pred)
# @tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
# decorator commented out because pred.numpy(): pred not evaluated yet.
def predict(self, inputs, labels):
labels = tf.squeeze(labels)
pred = self.model([inputs], training=False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment