pytorch userwarning记录

本文解决了PyTorch中masked_fill_函数使用uint8类型mask的警告问题,建议改为bool类型,避免未来版本的兼容性问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

userwarning描述:/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.

解决方法:

masked_fill_(data)

改成

masked_fill_(data.bool())

即可

G:\Project\pycharm_project\Pytorch_Experiment\.venv\Scripts\python.exe G:\Project\pycharm_project\Pytorch_Experiment\Mario\Mario.py G:\Project\pycharm_project\Pytorch_Experiment\Mario\Mario.py:163: SyntaxWarning: invalid escape sequence '\g' """计算时间差分目标值[^1]$$r + \gamma \max_{a}Q_{\text{target}}(s_{t+1},a)$$""" G:\Project\pycharm_project\Pytorch_Experiment\Mario\Mario.py:170: SyntaxWarning: invalid escape sequence '\e' """更新探索率[^1]$$\epsilon = \max(\epsilon_{\text{min}}, \epsilon \times \text{decay})$$""" G:\Project\pycharm_project\Pytorch_Experiment\.venv\Lib\site-packages\gym\envs\registration.py:593: UserWarning: WARN: The environment SuperMarioBros-1-1-v0 is out of date. You should consider upgrading to version `v3`. logger.warn( Traceback (most recent call last): File "G:\Project\pycharm_project\Pytorch_Experiment\Mario\Mario.py", line 303, in <module> train() File "G:\Project\pycharm_project\Pytorch_Experiment\Mario\Mario.py", line 229, in train env = gym_super_mario_bros.make("SuperMarioBros-1-1-v0") ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "G:\Project\pycharm_project\Pytorch_Experiment\.venv\Lib\site-packages\gym\envs\registration.py", line 662, in make env = env_creator(**_kwargs) ^^^^^^^^^^^^^^^^^^^^^^ File "G:\Project\pycharm_project\Pytorch_Experiment\.venv\Lib\site-packages\gym_super_mario_bros\smb_env.py", line 52, in __init__ super(SuperMarioBrosEnv, self).__init__(rom) File "G:\Project\pycharm_project\Pytorch_Experiment\.venv\Lib\site-packages\nes_py\nes_env.py", line 126, in __init__ _ = rom.prg_rom ^^^^^^^^^^^ File "G:\Project\pycharm_project\Pytorch_Experiment\.venv\Lib\site-packages\nes_py\_rom.py", line 204, in prg_rom return self.raw_data[self.prg_rom_start:self.prg_rom_stop] ^^^^^^^^^^^^^^^^^ File "G:\Project\pycharm_project\Pytorch_Experiment\.venv\Lib\site-packages\nes_py\_rom.py", line 198, in prg_rom_stop return self.prg_rom_start + self.prg_rom_size * 2**10 ~~~~~~~~~~~~~~~~~~^~~~~~~ OverflowError: Python integer 1024 out of bounds for uint8
最新发布
06-06
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值