Skip to content

Instantly share code, notes, and snippets.

@rjpower

rjpower/tpu_reduced_thread.py

Created January 20, 2019 20:54
Show Gist options
  • Save rjpower/8c9723e99ed72c61e768bfede9630c2e to your computer and use it in GitHub Desktop.
Save rjpower/8c9723e99ed72c61e768bfede9630c2e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tpu_reduced_thread.py
session_config = tf.ConfigProto(
intra_op_parallelism_threads=64,
inter_op_parallelism_threads=4,
)
...
run_config = tpu.RunConfig(
session_config=session_config,
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment