search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.88k stars 587 forks source link

Mixed precision training #329

Open metachao opened 3 years ago

metachao commented 3 years ago

Hello, how to use the apex module to do mixed precision training? Thanks

WongKinYiu commented 3 years ago

apex already native integrate in pytorch version > 1.6. default setting will use mixed precision training.