训练时，gpu占用并不满，但是增加batch_size会提示显存溢出，请问这是什么问题？

bubbliiiing / yolov4-tf2

这是一个yolo4-tf2（tensorflow2）的源码，可以用于训练自己的模型。

MIT License

314 stars 111 forks source link

Closed Mulbetty closed 3 years ago

Mulbetty commented 3 years ago

是我gpu型号的问题吗？

bubbliiiing commented 3 years ago

你这看起来都没用GPU，环境对吗

Mulbetty commented 3 years ago

环境和你的有点差别，不过应该没问题。我改用你的torch版本了

bubbliiiing commented 3 years ago