1:Warning: masked_scatter_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead
找到需要变换类型的训练集变量。这里要求训练集mask是bool类型而不是unit8,在返回训练集的时候加上 .bool()即可。
posted on 2022-05-12 16:06 Y-flower 阅读(107) 评论(0) 收藏 举报