F.cross entropy vs torch.nn.Cross_Entropy_Loss

6

You are not using a weighted cross entropy. So no difference as mentionned above.

4 Likes