网络传入的类型错误,我一开始是expected scalar type Floatbut found long
train_tensor = train_tensor.to(torch.float32)
然后就报
RuntimeError: expected scalar type Double but found Float
然后直接网络哪里修改

然后就好了开始训练

RuntimeError: expected scalar type Double but found Float
阅读 77
2022-12-09
网络传入的类型错误,我一开始是expected scalar type Floatbut found long
train_tensor = train_tensor.to(torch.float32)
然后就报
RuntimeError: expected scalar type Double but found Float
然后直接网络哪里修改

然后就好了开始训练

相关推荐
精彩评论(0)