longcw / yolo2-pytorch

YOLOv2 in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

memory usage

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