@ptrblck works perfectly! By any chance do you anyone who does similar things so that I can look at what hyper parameters and architecture they are using? Read all of cs231n was super interesting thank you for that advice last week.