Created
July 25, 2019 20:59
-
-
Save julled/b45f1f63b581103a966bda4aeda6fad9 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Change Line 62 in "model_main.py" from | |
config = tf.estimator.RunConfig(model_dir=FLAGS.model_dir) | |
to | |
config = tf.ConfigProto() | |
config.gpu_options.allow_growth = True | |
config.gpu_options.per_process_gpu_memory_fraction = 0.1 | |
config = tf.estimator.RunConfig(model_dir=FLAGS.model_dir, | |
session_config=config) | |
this way it should minimize your memory needs |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment