Limit Memory in Process of Training

Vann Ponlork
Sep 14, 2019 · 1 min read

To limit memory in training process avoid OOM

In training process, there are frequently meet the problem out of memory , so we should limit the memory to what we want, for me I limit memory to 80% of my GPU.

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.8
set_session(tf.Session(config=config))

LASTMILE WORKS / DYNAMO TECH - R&D Project

Developing next-generation technology in Combodia

Medium is an open platform where 170 million readers come to find insightful and dynamic thinking. Here, expert and undiscovered voices alike dive into the heart of any topic and bring new ideas to the surface. Learn more

Follow the writers, publications, and topics that matter to you, and you’ll see them on your homepage and in your inbox. Explore

If you have a story to tell, knowledge to share, or a perspective to offer — welcome home. It’s easy and free to post your thinking on any topic. Write on Medium

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store