From f7e3cfceb1932d41e628969060200804fc50c271 Mon Sep 17 00:00:00 2001
From: Connor Olding <cloningdonor@gmail.com>
Date: Thu, 5 Jan 2017 04:56:35 -0800
Subject: [PATCH] .

---
 resnet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/resnet.py b/resnet.py
index 1e4585f..bd90b32 100755
--- a/resnet.py
+++ b/resnet.py
@@ -52,7 +52,7 @@ use_image_generator = True
 def prepare(X, y):
     X = X.reshape(X.shape[0], 1, width, height).astype('float32') / 255
     # convert class vectors to binary class matrices
-    Y = to_categorical(y_train, nb_classes)
+    Y = to_categorical(y, nb_classes)
     return X, Y
 
 # the data, shuffled and split between train and test sets