search

airlab-unibas / airlab

Image registration laboratory for 2D and 3D image data
Apache License 2.0
408 stars 92 forks source link

Half-Precision Training #30

Open jlevy44 opened 4 years ago

jlevy44 commented 4 years ago

Can we use float16 using this framework? Has there been consideration for allowing integration with nvidia apex: https://github.com/NVIDIA/apex

Thanks!

RobinSandkuehler commented 4 years ago

Hi, since the type of each image could be defined in the beginning. It should be possible. The integration of apex is definitely a very interesting idea which we will consider in the future.

jlevy44 commented 3 years ago

https://pytorch.org/docs/stable/amp.html