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