memory usage
riyaj8888 opened this issue · comments
Hurizma1003 commented
when run testing code on single image , it uses full cpu of 16 gpu ram and then switch to gpu.. can move that part which is running on cpu to gpu ..please help
thanks
YOLOv2 in PyTorch
riyaj8888 opened this issue · comments
when run testing code on single image , it uses full cpu of 16 gpu ram and then switch to gpu.. can move that part which is running on cpu to gpu ..please help
thanks