mmdection训练出现变量定义错误

刘啊咚咚锵

于 2022-02-08 12:47:33 发布

阅读量2.6k

点赞数 3

CC 4.0 BY-SA版权

分类专栏：笔记深度学习文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/liu_dongd/article/details/122821627

笔记同时被 2 个专栏收录

3 篇文章

订阅专栏

深度学习

3 篇文章

订阅专栏

在使用mmdetection进行模型训练时遇到UnboundLocalError，原因是torch1.8.xx的bug。解决方案是手动修改adamw.py文件，将报错行前的空格删除。此问题在后续的torch版本已修复。

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

出现问题

使用mmdetection训练模型，出现下面错误
错误截图
错误提示是这样的

UnboundLocalError: local variable 'beta1' referenced before assignment

问题原因

使用了torch1.8.xx构建环境，这似乎是torch1.8.xx下面的一个bug。在后续的torch版本已经修复。

解决方法

找到报错的路径下的adamw.py文件，在上面报错的图上倒数第三行会有，我这里是

.conda/envs/liudong-mmlab/lib/python3.7/site-packages/torch/optim/adamw.py

打开该路径下的adamw.py文件，找到第110行

 F.adamw(params_with_grad,

将改行前面空格删掉一个，到下面的程度

            beta1, beta2 = group['betas']
                # update the steps for each param group update
                state['step'] += 1
                # record the step after step update
                state_steps.append(state['step'])

        F.adamw(params_with_grad,
                    grads,
                    exp_avgs,
                    exp_avg_sqs,
                    max_exp_avg_sqs,
                    state_steps,