From 9a02bec1be747ebefc15b44a0ddbd0569207b0fe Mon Sep 17 00:00:00 2001 From: Marc Lelarge Date: Fri, 23 Feb 2024 16:51:27 +0100 Subject: [PATCH] typo loss --- modules/3-loss-functions-for-classification.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/3-loss-functions-for-classification.md b/modules/3-loss-functions-for-classification.md index 5397a96..cb5f76b 100644 --- a/modules/3-loss-functions-for-classification.md +++ b/modules/3-loss-functions-for-classification.md @@ -58,7 +58,7 @@ loss1 = nn.NLLLoss() loss2 = nn.CrossEntropyLoss() C = 8 input = torch.randn(3,C,4,5) -target = torch.empty(3,4,5 dtype=torch.long).random_(0,C) +target = torch.empty(3,4,5, dtype=torch.long).random_(0,C) assert loss1(m(input),target) == loss2(input,target) ```