Thanks! Yeah I had initialized my loss function accordingly, just saved time pasting the code. I can double check the shapes on the input and targets but I believe there was no issue there as well. But if from a higher level view this should work the same, then thats all I needed to confirm.