Whether Yolov4-CSP and Yolov4-Large models are tested using FP32 or FP16 ?

WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network

GNU General Public License v3.0

2.02k stars 575 forks source link

Open mls1999725 opened 3 years ago

mls1999725 commented 3 years ago

Hi, I wonder whether Yolov4-CSP and Yolov4-Large models are tested using FP32 or FP16 ? Looking forward to your reply.

WongKinYiu commented 3 years ago

all results are tested using fp16 if the gpu has tensor core, otherwise fp32.

mls1999725 commented 3 years ago

all results are tested using fp16 if the gpu has tensor core, otherwise fp32.

Thanks.

mls1999725 commented 3 years ago

By the way, use model.half() to achieve testing by fp16, is that right ?

WongKinYiu commented 3 years ago

yes.