你好，请问在predict中，可以修改每次inference的batch_size大小么？

bubbliiiing / mobilenet-yolov4-pytorch

这是一个mobilenet-yolov4的库，把yolov4主干网络修改成了mobilenet，修改了Panet的卷积组成，使参数量大幅度缩小。

MIT License

369 stars 104 forks source link

Open fantow opened 2 years ago

fantow commented 2 years ago

期望可以通过增加inference的batch_size，来提高gpu利用率。

bubbliiiing commented 2 years ago

暂时还不太行……，得简单改改

fantow commented 2 years ago

暂时还不太行……，得简单改改

好的，谢谢