该错误是由于数据的类型不对导致的,
但是不要被提示误导,并不是需要转换为double
而应该转为float32
假设数据变量为data
则将其转为float32代码如下
data = data.to(torch.float32)
这样就可以了
该错误是由于数据的类型不对导致的,
但是不要被提示误导,并不是需要转换为double
而应该转为float32
假设数据变量为data
则将其转为float32代码如下
data = data.to(torch.float32)
这样就可以了