Open Tma2333 opened 2 years ago
torch.nn.CrossEntropyLoss expect you to pass raw, unnormalized logit, however, either logit or probability is passing into cross entropy loss. If the probability is used will lead to unintended result.
torch.nn.CrossEntropyLoss
torch.nn.CrossEntropyLossexpect you to pass raw, unnormalized logit, however, either logit or probability is passing into cross entropy loss. If the probability is used will lead to unintended result.