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

snapshot...

parent 87a67485
Branches
No related tags found
No related merge requests found
...@@ -607,7 +607,7 @@ class SRCNN: ...@@ -607,7 +607,7 @@ 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, axis=[3]), pred)
t_loss = self.loss(labels, pred) t_loss = self.loss(labels, pred)
self.test_labels.append(labels) self.test_labels.append(labels)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment