Comments (9)
@leJson Could you please specify where to find the parameter "n_cup" you're talking about? Thanks in advance
parser.add_argument("--n_cpu",dest='n_cpu',type=int,default=2,help="torch多线程核数")在这里把defualt=2改为defualt=0,也可以在训练的时候指定.
from yolov3-network-slimming.
Me,too,and did you solve it?
from yolov3-network-slimming.
Me,too,and did you solve it?
do not use "subdivisions", it does not work, so set "batch"
@cococener
from yolov3-network-slimming.
调整cfg文件中的batch和subvision参数
from yolov3-network-slimming.
When I change the batch number, it threw the following RunTimeError:
/usr/local/lib/python3.6/dist-packages/torch/nn/_reduction.py:15: UserWarning: reduction='elementwise_mean' is deprecated, please use reduction='mean' instead.
warnings.warn("reduction='elementwise_mean' is deprecated, please use reduction='mean' instead.")
Traceback (most recent call last):
File "sparsity_train.py", line 152, in
train()
File "sparsity_train.py", line 98, in train
loss = model(imgs, targets)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/home/abanoub/Desktop/yolov3-network-slimming-master/yolomodel.py", line 332, in forward
x = torch.cat((map1, map2), 1)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 26 and 24 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71
Architecture is yolov3-tiny
from yolov3-network-slimming.
我在docker里训练也遇到了这个问题,解决办法是 n_cup设为0
from yolov3-network-slimming.
貌似原因是docker与pytorch之间的兼容问题
from yolov3-network-slimming.
@leJson Could you please specify where to find the parameter "n_cup" you're talking about? Thanks in advance
from yolov3-network-slimming.
@leJson it is rigth, follow this suggestion, it work. But there is another problem, in parse_config.py file, as the default, try to set options['gpus'] = '0, 1' let two GPU work together, but actually, when run python sparsity_train.py, it is only 1 gpu work, not 2 together, any advise?
from yolov3-network-slimming.
Related Issues (20)
- 稀疏化训练时出现这个错,没明白原因。希望大佬给点建议RuntimeError: cannot perform reduction function max HOT 10
- 稀疏化训练过程中,电脑重启!!!有遇到类似问题的么 求解决方案
- About multi GPUs used
- 关于在yolov2上使用的问题?
- 尝试多GPU训练后loss上涨
- 稀疏训练指标异常 HOT 1
- 关于稀疏化训练基础网络的问题 HOT 1
- 关于donntprune的一个疑问
- 稀疏化训练速度非常缓慢
- 运行出错python sparsity_train.py -sr --s 0.0001 --image_folder obj.data --cfg yolov3.cfg --weights yolov3.weights
- 稀疏化训练异常:打印出的数据 w和 h从某刻开始变得异常大 HOT 1
- train过程中w和h变得很大
- 稀疏训练自己的网络
- 关于训练问题
- 测试demo
- 稀疏训练 tensor not match
- 剪枝之后的模型预测不出来任何东西 HOT 2
- yolov3-VOC
- x, losses = self.module_list[i][0](x, targets) HOT 1
- Your pruned weights are not working.
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yolov3-network-slimming.