对于交叉熵损失，模型的输出类型需要为torch.float32,而标签的类型也需要为torch.float32
- 文章链接: http://hust-keep-lin.github.io/2024/04/07/cross-entropy的输出及标签/
- 版权声明: 本网站所有文章除特别声明外,均采用 CC BY-NC-ND 4.0 许可协议。转载请注明出处!
对于交叉熵损失，模型的输出类型需要为torch.float32,而标签的类型也需要为torch.float32
