Torchsummary должен обратить внимание на то, передается ли модель в cuda, и она должна соответствовать параметру устройства.
NoSuchKey
рекомендация
отblog.csdn.net/LIWEI940638093/article/details/126722031
рекомендация
ранжирование