Torchsummary should pay attention to whether the model is passed to cuda, and it must be consistent with the device parameter

NoSuchKey

Guess you like

Origin blog.csdn.net/LIWEI940638093/article/details/126722031