I just edited it to include the loss function.  I prefer LogSoftmax but that caused an argument error on the forward method.
But I’m loading my own dataset.   This was code for mnist.
It seems I should probably just upgrade everything to start.