Torchsummary должен обратить внимание на то, передается ли модель в cuda, и она должна соответствовать параметру устройства.

NoSuchKey

рекомендация

отblog.csdn.net/LIWEI940638093/article/details/126722031
рекомендация