#报错记录#RuntimeError: Expected tensor for 'out' to have the same device as tensor for argument #2 'mat
最后发现是device_id没有全部修改成0 ...
花费 5 ms
#报错记录#TypeError: __init__() takes 1 positional argument but 2 were given
class conv_try(nn.Module): def __init__(self): super(conv_try, self).__init__() ...