AttributeError: module ‘torch‘ has no attribute ‘bool‘

AttributeError: module ‘torch’ has no attribute ‘bool’

问题描述：

使用pytorch 1.0.0 的时候，出现该错误

问题处理：

原因是因为：版本问题。我用0.4.0版本测试发现torch.bool是有的，但是1.0.0确实没有了

解决：解决办法就是把torch.bool换成torch.uint8

修改前的代码：

im_mask = torch.zeros((im_h, im_w), dtype=torch.bool)

修改后的代码：

im_mask = torch.zeros((im_h, im_w), dtype=torch.uint8)