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)