From 5eb0069236af038e3bb7d4f02bbee5d53a095ac7 Mon Sep 17 00:00:00 2001 From: Heiko J Schick Date: Thu, 22 Oct 2020 16:45:07 +0200 Subject: [PATCH] Re-enabled training --- lenet5.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/lenet5.py b/lenet5.py index 6ecd241..a494ee0 100644 --- a/lenet5.py +++ b/lenet5.py @@ -164,10 +164,6 @@ 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]) @@ -208,9 +204,9 @@ def main(): model = create_lenet5() # Step 8: - # Train and save the LeNet-5 model - # train_lenet5(model) - # model.save('lenet5.h5') + # Train, save and load the LeNet-5 model + train_lenet5(model) + model.save('lenet5.h5') model = load_model('lenet5.h5') # Step 9: