HybridNets
HybridNets copied to clipboard
Why does HybridNets require larger GPU memory?
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?