simple-faster-rcnn-pytorch icon indicating copy to clipboard operation
simple-faster-rcnn-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon chenyuntc

关于batch_size的设置

Open Smallwhite999 opened this issue 4 years ago • 3 comments

代码中的batch_size为什么只能设置为1啊？如何去修改为其它的值呢？

Smallwhite999 avatar Sep 16 '20 12:09 Smallwhite999

看显存大小吧

xiangweifeng avatar Oct 07 '20 08:10 xiangweifeng

因为里面很多代码都是针对一个batch写的，所以想用大于1的batch_size训练要自己修改代码

LGsmile avatar Mar 18 '21 01:03 LGsmile

batch_size是无法直接修改的，因为在data/dataset.py里定义的训练集Dataset将返回元素定义为img, bbox, label, scale而bbox的个数不同图片是不同的，维度不同的bbox无法合并成一个Tensor，这会引起torch.stack报错。需要去重新定义Dataset和读入图片及标签的逻辑才能修改batch_size。

fnsflm avatar Dec 12 '23 19:12 fnsflm