In the training portion I needed to incorporate a “.float()” command to inputs and labels to make this work, not sure if that is generic issue or specific to my dataset.

inputs = Variable(torch.from_numpy(x_train).float())
labels = Variable(torch.from_numpy(y_correct).float())

Also note that the relevant imports may be appropriate to feature in this tutorial, otherwise for anyone interested they are listed in the linked github repo.

Writing for fun and because it helps me organize my thoughts. I also write software to prepare data for machine learning at automunge.com

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store