bubbliiiing/yolov4-tiny-pytorch

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

JohnFiv opened this issue · 5 comments

JohnFiv commented

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

bubbliiiing commented

emm我没改代码呀0 0

dyhisbig commented

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

bubbliiiing commented

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

ChenMaolong commented

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

bubbliiiing commented

1.2 1.7都可以