From 873043b9fb3efecdff0fbdbb5dd95cbd66efd7f5 Mon Sep 17 00:00:00 2001 From: Willem Melching Date: Wed, 2 Sep 2020 15:12:30 +0200 Subject: [PATCH] 1024 MB should be enough for anyone --- selfdrive/modeld/runners/keras_runner.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/selfdrive/modeld/runners/keras_runner.py b/selfdrive/modeld/runners/keras_runner.py index 0a9eddeac7..f38902b7c7 100755 --- a/selfdrive/modeld/runners/keras_runner.py +++ b/selfdrive/modeld/runners/keras_runner.py @@ -41,10 +41,9 @@ if __name__ == "__main__": # limit gram alloc gpus = tf.config.experimental.list_physical_devices('GPU') if len(gpus) > 0: - tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)]) + tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)]) m = load_model(sys.argv[1]) print(m, file=sys.stderr) run_loop(m) -