Train with lower precision

mingyuliutw / UNIT

Unsupervised Image-to-Image Translation

Other

1.98k stars 360 forks source link

Train with lower precision #107

Closed mhusseinsh closed 3 years ago

mhusseinsh commented 5 years ago

Hello, I would like to ask if there is a way to train the UNIT model with a lower precision? just to to give it a try if I can result in reducing training/inference time

Which part do I need to edit in the code?

mingyuliutw commented 3 years ago

The updated UNIT implementation is available at

https://github.com/NVlabs/imaginaire

We have low precision training.

Support should be better there.