search

bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码，可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

为什么现在训练batch不能设置为以前那么大了，GPU显存才占不到一半就报显存溢出了。 #87

Open JohnFiv opened 2 years ago

JohnFiv commented 2 years ago

我是12G显存，最新的代码每次运行到一半显存就开始报显存不够

bubbliiiing commented 2 years ago

emm我没改代码呀0 0

dyhisbig commented 2 years ago

估计是代码哪里的问题，往显存中加载数据，但是显存溢出，所以就是只占用了一半的显存，你试试batch改小点

bubbliiiing commented 2 years ago

嗯？我6G都能batch 32，我觉得是环境配置的问题

ChenMaolong commented 2 years ago

嗯？我6G都能batch 32，我觉得是环境配置的问题 大佬你batch 32下的环境pytorch是什么版本啊，torch1.8?

bubbliiiing commented 2 years ago

1.2 1.7都可以