What’s the advantage of doing this? This is the way I originally tried to implement it, but I couldn’t work out the accuracy or loss implementation. I’ll try doing it this way and see what happens 
Also, would you recommend CrossEntropyLoss, NLLloss or some other function for this sort of thing, and why one over the other?