pytorch 报错：RuntimeError: cuda runtime error (59) : device-side assert triggered（已解决）

最新推荐文章于 2024-07-08 08:30:00 发布

贝利亚黄昏

最新推荐文章于 2024-07-08 08:30:00 发布

阅读量1.5k

点赞数

分类专栏：图像处理技能技巧文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_39351946/article/details/121356817

版权

这篇博客解决了在使用PyTorch进行深度学习时遇到的RuntimeError: cuda runtime error (59)，问题根源在于模型输出通道数与分类类别数不匹配。此外，还提到了训练数据标签错误，包括超出分类数目和包含0的情况，指出在PyTorch中0及以下类别标签会导致错误，解决方案是确保标签从1开始并对应正确的分类数量。

摘要由CSDN通过智能技术生成

======> Epoch 0 starting train.
Traceback (most recent call last):
  File "train_exbase.py", line 326, in <module>
    main(args, logger)
  File "train_exbase.py"