因为我有数学物理背景,所以清楚卷积的原理。但是在看pytorch文档的时候感到非常头大,罗列的公式以及各种令人眩晕的下标让入门新手不知所云...最初我以为torch.nn.conv1d的参数in_channel/out_channel表示图像的通道数,经过运行错误提示之后,才知道[in_channel,kernel_size]构成了卷积核。
loss函数中target/label的类型必须是dtype=torch.long 。
在使用cross_entropy时出现了一个错误RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1),查看functional.py的1550行中:
其中log_softmax(input, 1)指定了dim=1,而log_softmax实现如下:
排查之后发现,cross_entropy输入的矩阵行数需要等于taget/label的个数,即行对行。