交叉熵需要传入一个output和一个target。nn.CrossEntropyLoss(output, target)
。
其中:
output.dtype : torch.FloatTorch
target.dtype : torch.LongTorch
对数据和标签做如下处理:
x = torch.from_numpy(x).float()
target = torch.from_numpy(target).long()
参考:
交叉熵需要传入一个output和一个target。nn.CrossEntropyLoss(output, target)
。
其中:
output.dtype : torch.FloatTorch
target.dtype : torch.LongTorch
对数据和标签做如下处理:
x = torch.from_numpy(x).float()
target = torch.from_numpy(target).long()
参考: