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

snapshot..

parent b71bdab5
No related branches found
No related tags found
No related merge requests found
...@@ -605,8 +605,8 @@ class ESPCN: ...@@ -605,8 +605,8 @@ class ESPCN:
self.reset_test_metrics() self.reset_test_metrics()
for data0, data1, label in self.test_dataset: for data, label in self.test_dataset:
ds = tf.data.Dataset.from_tensor_slices((data0, data1, label)) ds = tf.data.Dataset.from_tensor_slices((data, label))
ds = ds.batch(BATCH_SIZE) ds = ds.batch(BATCH_SIZE)
for mini_batch_test in ds: for mini_batch_test in ds:
self.predict(mini_batch_test) self.predict(mini_batch_test)
...@@ -639,7 +639,7 @@ class ESPCN: ...@@ -639,7 +639,7 @@ class ESPCN:
self.reset_test_metrics() self.reset_test_metrics()
pred = self.model([data]) pred = self.model([data], training=False)
self.test_probs = pred self.test_probs = pred
pred = pred.numpy() pred = pred.numpy()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment