理解Pytorch里面nn.CrossEntropyLoss的含义
理解Pytorch里面nn.CrossEntropyLoss的含义先说nn.CrossEntropyLoss的参数,如果神经网络的输出output是一个(batch_size, num_class, h, w)的tensor(其中,num_class代表分类问题的类别数,h为图像高度,w为图像宽度),则nn.CrossEntropyLoss需要的label形状为(batch_size, h, ...
原创
2020-03-25 23:59:39 ·
5072 阅读 ·
0 评论