当前位置:网站首页>batchnorm. Py this file single GPU operation error solution
batchnorm. Py this file single GPU operation error solution
2022-07-05 01:26:00 【Zi Yan Ruoshui】
The original
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
Modify file :/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的采用入口?
- [wave modeling 3] three dimensional random real wave modeling and wave generator modeling matlab simulation
- 微信小程序:微群人脉微信小程序源码下载全新社群系统优化版支持代理会员系统功能超高收益
- 【海浪建模3】三维随机真实海浪建模以及海浪发电机建模matlab仿真
- Behind the cluster listing, to what extent is the Chinese restaurant chain "rolled"?
- Express routing, express middleware, using express write interface
- [CTF] AWDP summary (WEB)
- What is the current situation and Prospect of the software testing industry in 2022?
- Daily question brushing record (13)
- 微信小程序:全网独家小程序版本独立微信社群人脉
猜你喜欢
How to use words to describe breaking change in Spartacus UI of SAP e-commerce cloud
Take you ten days to easily complete the go micro service series (IX. link tracking)
Detailed explanation of multi-mode input event distribution mechanism
Great God developed the new H5 version of arXiv, saying goodbye to formula typography errors in one step, and mobile phones can also easily read literature
[wave modeling 2] three dimensional wave modeling and wave generator modeling matlab simulation
Implementation steps of master detail detail layout mode of SAP ui5 application
JS implementation determines whether the point is within the polygon range
【CTF】AWDP总结(Web)
Applet live + e-commerce, if you want to be a new retail e-commerce, use it!
增量备份 ?db full
随机推荐
Call Huawei order service to verify the purchase token interface and return connection reset
[pure tone hearing test] pure tone hearing test system based on MATLAB
【海浪建模3】三维随机真实海浪建模以及海浪发电机建模matlab仿真
微信小程序:最新wordpress黑金壁纸微信小程序 二开修复版源码下载支持流量主收益
[FPGA tutorial case 9] design and implementation of clock manager based on vivado core
Pandora IOT development board learning (RT thread) - Experiment 4 buzzer + motor experiment [key external interrupt] (learning notes)
LeetCode周赛 + AcWing周赛(T4/T3)分析对比
潘多拉 IOT 开发板学习(RT-Thread)—— 实验4 蜂鸣器+马达实验【按键外部中断】(学习笔记)
[flutter topic] 64 illustration basic textfield text input box (I) # yyds dry goods inventory #
Roads and routes -- dfs+topsort+dijkstra+ mapping
node工程中package.json文件作用是什么?里面的^尖括号和~波浪号是什么意思?
微信小程序:全网独家小程序版本独立微信社群人脉
dotnet-exec 0.6.0 released
小程序直播 + 电商,想做新零售电商就用它吧!
Chia Tai International Futures: what is the master account and how to open it?
Senior Test / development programmers write no bugs? Qualifications (shackles) don't be afraid of mistakes
Query for Boolean field as "not true" (e.g. either false or non-existent)
PHP wechat official account development
[development of large e-commerce projects] performance pressure test - Performance Monitoring - heap memory and garbage collection -39
【CTF】AWDP总结(Web)