CrossEntropyLoss

设备端断言触发，是哪里出了问题吗？

设备端断言触发，是哪里出了问题吗？一言难尽。 一开始我还以为是显存不够了毕竟我的模型比较大，数据量也比较多。于是我把batch size调小了一半，后来啊还是不行。后来又尝试了各种方法，比如清理CUDA缓存、梗新驱动程序、重新安装PyTor