how to change batch_size in vedadet?

Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch

Apache License 2.0

498 stars 128 forks source link

Open saehan-choi opened 2 years ago

saehan-choi commented 2 years ago

I'm customizing it but cuda memory error occured please let me know..

gd1925 commented 2 years ago

Hi Saehan-choi, Did you find out the answer to this? Even I need to change it.. Thank you.