pytorch报错:

RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:224

原因:模型输出的类别个数和读入的类别个数不一致导致的

将输出类别个数修改为与输入类别个数一致即可。

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐