HybridNets icon indicating copy to clipboard operation
HybridNets copied to clipboard

Published 20 hours ago • verified publisher icon datvuthanh

Why does HybridNets require larger GPU memory?

Open yaoshanliang opened this issue 1 year ago • 1 comments

In YOLOP, I can set batch_size to 64 in a 3090 sever. However, in HybridNets, the batch_size can only be set to 8. Otherwise, the program shows the message "Out of Memory".

I have made some modifications according to the readme file and some issues in this project, including python train.py --cal_map False, python train.py --conf_thres 0.5, but it still trains slowly. Do I need to modify some other configs in your project?

yaoshanliang avatar Feb 25 '23 13:02 yaoshanliang