issues
search
bubbliiiing
/
mobilenet-yolov4-pytorch
这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
MIT License
369
stars
104
forks
source link
你好,请问在predict中,可以修改每次inference的batch_size大小么?
#18
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
暂时还不太行……,得简单改改
好的,谢谢
期望可以通过增加inference的batch_size,来提高gpu利用率。