Not at all, thanks for the feedback. Fair enough, but actually I am testing various loss functions in order to reveal if I might improve the accuracy. So, when using  nn.CrossEntropyLoss, should I directly return the output of the FC layers as the last command of the forward function?