How to train model with Automatic Mixed Precision?

valeoai / WoodScape

The repository containing tools and information about the WoodScape dataset.

https://woodscape.valeo.com/dataset

610 stars 124 forks source link

How to train model with Automatic Mixed Precision? #78

Open Splendon opened 2 years ago

Splendon commented 2 years ago

Hi, thanks very much for this excellent work. Since the gpu resource is expansive, could you release the code with amp(Automatic Mixed Precision)? Thanks