问题解决

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)    收藏  举报

导航