/pytorch/aten/src/ATen/native/cuda/LegacyDefinitions.cpp:14: UserWarning: masked_fill_ received a ma

最新推荐文章于 2019-12-05 23:03:01 发布

原创最新推荐文章于 2019-12-05 23:03:01 发布 · 5.2k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

疑难问题

当前文章被收录于：

疑难问题

9 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

出现如下异常:

/pytorch/aten/src/ATen/native/cuda/LegacyDefinitions.cpp:14: UserWarning: masked_fill_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead.

The same problem. Fixed it by changing every
"mask = blahblahblah.byte()"
to
"mask = blahblahblah.bool()"
in three files: main.py, main_parse.py and model/crf.py