From 24778be577548a0d7fe234f99e64ee45a8e42c13 Mon Sep 17 00:00:00 2001
From: Heiko J Schick <heiko.schick@huawei.com>
Date: Thu, 22 Oct 2020 14:49:27 +0200
Subject: [PATCH] Minor changes

---
 lenet5.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lenet5.py b/lenet5.py
index 7ed8728..6ecd241 100644
--- a/lenet5.py
+++ b/lenet5.py
@@ -164,6 +164,10 @@ def main():
     test['features'], test['labels'] = read_mnist('t10k-images-idx3-ubyte.gz',
                                                   't10k-labels-idx1-ubyte.gz')
 
+    print(type(train))
+    print(type(train['features']))
+    print(type(train['labels']))
+
     # Step 3:
     # Explore the dataset
     print('Number of training images:', train['features'].shape[0])