search

funkelab / gunpowder

A library to facilitate machine learning on multi-dimensional images.
https://funkelab.github.io/gunpowder/
MIT License
78 stars 56 forks source link

Auto mixed precision for tensorflow #128

Closed abred closed 3 years ago

abred commented 4 years ago

Switch to turn on automatic mixed precision (rtx 2080ti and up) automatically checks if tensors are (numerically) save to be transformed to float16 instead of float32.