diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 10d32ca21954aeeeaaff8a516b584f0f2c3705a8..02c30c6bfadc4c0ea3484cf6f9f27f282e5b3d4e 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -518,7 +518,7 @@ class SRCNN: def predict(self, mini_batch): inputs = [mini_batch[0]] labels = mini_batch[1] - pred = self.model(inputs, training=False) + pred = self.model([inputs], training=False) t_loss = self.loss(labels, pred) self.test_labels.append(labels)