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

snapshot...

parent f8e91300
Branches
No related tags found
No related merge requests found
...@@ -612,7 +612,8 @@ class SRCNN: ...@@ -612,7 +612,8 @@ class SRCNN:
# decorator commented out because pred.numpy(): pred not evaluated yet. # decorator commented out because pred.numpy(): pred not evaluated yet.
def predict(self, inputs, labels): def predict(self, inputs, labels):
pred = self.model([inputs], training=False) pred = self.model([inputs], training=False)
t_loss = self.loss(tf.squeeze(labels), pred) # t_loss = self.loss(tf.squeeze(labels), pred)
t_loss = self.loss(labels, pred)
self.test_labels.append(labels) self.test_labels.append(labels)
self.test_preds.append(pred.numpy()) 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