代码如下:
Loss = torch.nn.CrossEntropyLoss()
loss = Loss(out, label)
修改的时候,直接在label后面加上.long(),如下所示:
Loss = torch.nn.CrossEntropyLoss()
loss = Loss(out, label.long())
代码如下:
Loss = torch.nn.CrossEntropyLoss()
loss = Loss(out, label)
修改的时候,直接在label后面加上.long(),如下所示:
Loss = torch.nn.CrossEntropyLoss()
loss = Loss(out, label.long())