可能是在使用交叉熵损失函数的时候,target需要是整数,才能转化成索引值,进而进行one-hot编码。
输出一下target的张量,可以看到每个值都后面有一个点.比如5.这样,应该表示的就是浮点类型的值。
这个时候需要target=target.long()执行一下类型转换。
可能是在使用交叉熵损失函数的时候,target需要是整数,才能转化成索引值,进而进行one-hot编码。
输出一下target的张量,可以看到每个值都后面有一个点.比如5.这样,应该表示的就是浮点类型的值。
这个时候需要target=target.long()执行一下类型转换。