Skip to content

Adding automatic mixed precision training support - #113

Open
vinhngx wants to merge 3 commits into
google:masterfrom
vinhngx:dqn
Open

vinhngx wants to merge 3 commits into
google:masterfrom
vinhngx:dqn

Conversation

@vinhngx

@vinhngx vinhngx commented Jul 24, 2019

Copy link
Copy Markdown

Automatic Mixed Precision training on GPU for Tensorflow has been recently introduced:

https://medium.com/tensorflow/automatic-mixed-precision-in-tensorflow-for-faster-ai-training-on-nvidia-gpus-6033234b2540

This PR adds GPU automatic mixed precision training to dopamine via setting a single OS flag:

export TF_ENABLE_AUTO_MIXED_PRECISION=1

@googlebot googlebot added the cla: yes CLA has been signed. label Jul 24, 2019

This branch has not been deployed

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

Labels

cla: yes CLA has been signed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants