fp16 or mixed-precision computation

LumenPallidium / neural-file-sorter

A neural network based file sorter. Trains an autoencoder to sort images or audio based on the similarity of their encodings, or uses the OpenAI CLIP model.

MIT License

27 stars 1 forks source link

fp16 or mixed-precision computation #13

Open LumenPallidium opened 3 years ago

LumenPallidium commented 3 years ago

either add NVIDIA apex.amp or options to convert all params with .half()