torchsummary 默认是在cuda上运行，报错的，将device修改为。嘿，你能搜索到这个问题，说明你说一个认真学习的同学，这个问题的细节值得思考。，但是模型权重早CPU，两种没在统一设备上，所以报错。特别注意，上面说的是输入数据在。请仔细看后面的调试界面。

本质是索引idx越界了，这个错误容易发生在分类任务，类别的输出和模型的输出不匹配，例如模型输出类别为8，但是却索引到9的位置，导致报错。