Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight';

NoSuchKey

Guess you like

Origin blog.csdn.net/qq_41872271/article/details/105701128