Speedup: Using 'mixed_precision'

PathwayCommons / pathway-abstract-classifier

A tool to classify journal abstracts with pathway content.

MIT License

0 stars 1 forks source link

Speedup: Using 'mixed_precision' #42

Closed jvwong closed 2 years ago

jvwong commented 2 years ago

Setting to use mixed precision could improve performance on GPU:

# Activate AMP - This is optional and may improve performance
from tensorflow.keras.mixed_precision import experimental as mixed_precision
...

Notes

see https://www.tensorflow.org/guide/mixed_precision
See notebook example
Could cause problems when no GPU present

jvwong commented 2 years ago

This didn't seem to make any difference on the workstation (X.X.X.150) under load tests as in #41 (probably useful in training?) but anyways its there.