From 04a405545f065d5d5e5cee0ca5b2481ded0b0563 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 17:04:54 +0200 Subject: [PATCH] Disabled growing of GPU RAM Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 04a5a3e..77a1c9e 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -176,7 +176,7 @@ def _ssd_val(args: argparse.Namespace) -> None: from twomartens.masterthesis import ssd config = tf.ConfigProto() - config.gpu_options.allow_growth = True + config.gpu_options.allow_growth = False tf.enable_eager_execution(config=config) batch_size = 16 image_size = 300