From 83d0c11e914b400bd7623de25056178393850456 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 22 Sep 2021 10:22:31 +0200
Subject: [PATCH] [Driver] update mnist/cifar10 validation script for list inps
---
src/finn/qnn-data/templates/driver/validate.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/qnn-data/templates/driver/validate.py b/src/finn/qnn-data/templates/driver/validate.py
index 001744cba..1b29d4342 100644
--- a/src/finn/qnn-data/templates/driver/validate.py
+++ b/src/finn/qnn-data/templates/driver/validate.py
@@ -94,11 +94,11 @@ if __name__ == "__main__":
test_labels = test_labels.reshape(n_batches, bsize)
for i in range(n_batches):
- ibuf_normal = test_imgs[i].reshape(driver.ibuf_packed_device.shape)
+ ibuf_normal = test_imgs[i].reshape(driver.ibuf_packed_device[0].shape)
exp = test_labels[i]
driver.copy_input_data_to_device(ibuf_normal)
driver.execute_on_buffers()
- obuf_normal = np.empty_like(driver.obuf_packed_device)
+ obuf_normal = np.empty_like(driver.obuf_packed_device[0])
driver.copy_output_data_from_device(obuf_normal)
ret = np.bincount(obuf_normal.flatten() == exp.flatten())
nok += ret[0]
--
GitLab