Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save julled/b45f1f63b581103a966bda4aeda6fad9 to your computer and use it in GitHub Desktop.
Save julled/b45f1f63b581103a966bda4aeda6fad9 to your computer and use it in GitHub Desktop.
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