always RuntimeError: CUDA error: out of memory about yolov3-network-slimming HOT 9 OPEN

@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.