RaphaelMeudec/mixed_preciision.py

Created January 16, 2020 16:07

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/RaphaelMeudec/7c235a56eef5ff74f86498d1664f28f5.js"></script>
Save RaphaelMeudec/7c235a56eef5ff74f86498d1664f28f5 to your computer and use it in GitHub Desktop.

Download ZIP

Lines to turn float32 into mixed precision training

Raw

mixed_preciision.py

	policy = tf.keras.mixed_precision.experimental.Policy('mixed_float16')
	tf.keras.mixed_precision.experimental.set_policy(policy)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment