search

wufan-tb / yolo_slowfast

Yolov5+SlowFast: Realtime Action Detection Based on PytorchVideo
417 stars 52 forks source link

怎么让它在gpu上运行 #22

Open white1wind opened 1 year ago

white1wind commented 1 year ago

从哪里修改可以在gpu运行

wufan-tb commented 1 year ago

默认就是gpu运行的，可以通过--device修改