DiffusionDet icon indicating copy to clipboard operation
DiffusionDet copied to clipboard

Published 20 hours ago • verified publisher icon ShoufaChen

where is BATCHSIZE? 请问数据的dataload是不是不在主代码库里，放在detectron2里面？

Open Meoooww opened this issue 1 year ago • 3 comments

找了半天没有找到，有没有大佬说一下batch'size在哪里设置的？感谢感谢！

Meoooww avatar Mar 30 '23 08:03 Meoooww

请问是不是在detectron2/configs/common/data/coco.py, 这里有一个dataloader.train = .... total_batch_size =16 num_workers = 4如果是在这的话，为啥我改成 total_batch_size =12还是提示cuda out of memory，我显卡是3070不至于这么拉的吧？

Meoooww avatar Mar 30 '23 09:03 Meoooww

Hi,

https://github.com/ShoufaChen/DiffusionDet/blob/1efb36d4b0b721f2d509bc72656235a94d73ab02/configs/Base-DiffusionDet.yaml#L19

ShoufaChen avatar Mar 31 '23 01:03 ShoufaChen

Hi,

https://github.com/ShoufaChen/DiffusionDet/blob/1efb36d4b0b721f2d509bc72656235a94d73ab02/configs/Base-DiffusionDet.yaml#L19

太谢谢您了！我现在就来尝试一下

Meoooww avatar Mar 31 '23 09:03 Meoooww