search

jingtianyilong / YOLOv4-pytorch

This is a pytorch repository of YOLOv4 that can be trained with custom dataset.
31 stars 12 forks source link

whether use GPU? #1

Closed gll-sketch closed 3 years ago

gll-sketch commented 3 years ago

Hi, I'd like to know if this code uses GPU in the evaluation?

jingtianyilong commented 3 years ago

Yes. It loads all available GPU but uses only one GPU for calculation