当前位置:网站首页>batchnorm.py这个文件单GPU运行报错解决
batchnorm.py这个文件单GPU运行报错解决
2022-07-05 01:24:00 【子燕若水】
原始的
if self.training and self.track_running_stats:
self.num_batches_tracked = self.num_batches_tracked + 1
if self.momentum is None: # use cumulative moving average
exponential_average_factor = 1.0 / self.num_batches_tracked.item()
else: # use exponential moving average
exponential_average_factor = self.momentum
need_sync = self.training or not self.track_running_stats
if need_sync:
process_group = torch.distributed.group.WORLD
if self.process_group:
process_group = self.process_group
world_size = torch.distributed.get_world_size(process_group)
need_sync = world_size > 1
修改文件:/home/ubuntu/venvs/venv/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py
(venv) [email protected]:~/code/ICDAR2021_MFD$ python tools/train.py configs/gfl/gfl_s50_fpn_2x_coco.py --gpus 1 --work-dir /home/ubuntu/data
Traceback (most recent call last):
File "tools/train.py", line 8, in <module>
import mmcv
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/mmcv/__init__.py", line 4, in <module>
from .fileio import *
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/mmcv/fileio/__init__.py", line 4, in <module>
from .io import dump, load, register_handler
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/mmcv/fileio/io.py", line 4, in <module>
from ..utils import is_list_of, is_str
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/mmcv/utils/__init__.py", line 19, in <module>
import torch
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/torch/__init__.py", line 448, in <module>
from .functional import *
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/torch/functional.py", line 4, in <module>
import torch.nn.functional as F
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/torch/nn/__init__.py", line 1, in <module>
from .modules import *
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/torch/nn/modules/__init__.py", line 17, in <module>
from .batchnorm import BatchNorm1d, BatchNorm2d, BatchNorm3d, SyncBatchNorm
File "/home/ubuntu/venvs/venv/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 489
need_sync = False
边栏推荐
- POAP:NFT的采用入口?
- How to use words to describe breaking change in Spartacus UI of SAP e-commerce cloud
- 【海浪建模2】三维海浪建模以及海浪发电机建模matlab仿真
- Call Huawei order service to verify the purchase token interface and return connection reset
- Playwright之录制
- 微信小程序:独立后台带分销功能月老办事处交友盲盒
- JS implementation determines whether the point is within the polygon range
- [FPGA tutorial case 10] design and implementation of complex multiplier based on Verilog
- The server time zone value ‘� й ��� ʱ 'is unrecognized or representatives more than one time zone【
- Heartless sword English translation of Xi Murong's youth without complaint
猜你喜欢
小程序直播 + 电商,想做新零售电商就用它吧!
Implementation steps of master detail detail layout mode of SAP ui5 application
流批一体在京东的探索与实践
Arbitrum:二维费用
微信小程序:独立后台带分销功能月老办事处交友盲盒
整理混乱的头文件,我用include what you use
微信小程序:星宿UI V1.5 wordpress系统资讯资源博客下载小程序微信QQ双端源码支持wordpress二级分类 加载动画优化
各大主流编程语言性能PK,结果出乎意料
Wechat applet: independent background with distribution function, Yuelao office blind box for making friends
Grabbing and sorting out external articles -- status bar [4]
随机推荐
【FPGA教程案例9】基于vivado核的时钟管理器设计与实现
Four pits in reentrantlock!
Database performance optimization tool
node工程中package.json文件作用是什么?里面的^尖括号和~波浪号是什么意思?
微信小程序:全网独家小程序版本独立微信社群人脉
How to safely eat apples on the edge of a cliff? Deepmind & openai gives the answer of 3D security reinforcement learning
Postman automatically fills headers
Using openpyxl module to write the binary list into excel file
107. Some details of SAP ui5 overflow toolbar container control and resize event processing
POAP:NFT的采用入口?
無心劍英譯席慕容《無怨的青春》
To sort out messy header files, I use include what you use
Yyds dry goods inventory kubernetes management business configuration methods? (08)
Actual combat simulation │ JWT login authentication
Playwright recording
Wechat applet: new independent backstage Yuelao office one yuan dating blind box
Database postragesq role membership
Chia Tai International Futures: what is the master account and how to open it?
微信小程序;胡言乱语生成器
RB technology stack