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

snapshot..

parent d80f3440
No related branches found
No related tags found
Loading
......@@ -614,19 +614,6 @@ class ESPCN:
print('loss, acc: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy())
# labels = np.concatenate(self.test_labels)
# self.test_labels = labels
#
# preds = np.concatenate(self.test_preds)
# self.test_probs = preds
#
# if NumClasses == 2:
# preds = np.where(preds > 0.5, 1, 0)
# else:
# preds = np.argmax(preds, axis=1)
#
# self.test_preds = preds
def do_evaluate(self, nda_lr, param, ckpt_dir):
ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment