欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

RuntimeError: Integer division of tensors using div or / is no longer supported, and in a future rel

程序员文章站 2022-03-04 14:09:39
...
RuntimeError: Integer division of tensors using div or / is no longer supported, and in a future release div will perform true division as in Python 3. Use true_divide or floor_divide (// in Python) instead.

在高版本的pytorch中,会报这个错误。
例子:

norm_img = transforms.Normalize((10,10,10), (1,1,1))(img)为1
print(norm_img)

报上面错误

解决方案:

对数据进行规范化处理

img=img.float()  #加这一行
norm_img = transforms.Normalize((10,10,10), (1,1,1))(img) 
print(norm_img)
相关标签: 报错解决