From 6a8e87e0190f8580e2acc81336070cfdab1e928b Mon Sep 17 00:00:00 2001 From: Andrew Sidhu Date: Sat, 18 Apr 2020 03:39:26 -0400 Subject: [PATCH] fix compile warning --- selfdrive/modeld/runners/keras_runner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/selfdrive/modeld/runners/keras_runner.py b/selfdrive/modeld/runners/keras_runner.py index 728fb20df6bea7..3f3a13d4787fa7 100755 --- a/selfdrive/modeld/runners/keras_runner.py +++ b/selfdrive/modeld/runners/keras_runner.py @@ -47,9 +47,9 @@ def run_loop(m,name): m.load_weights(f"{os.path.splitext(sys.argv[1])}.weights.keras") else: tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=256)]) - m = load_model(sys.argv[1]) + m = load_model(sys.argv[1], compile=False) else: - m = load_model(sys.argv[1]) + m = load_model(sys.argv[1], compile=False) bs = [int(np.product(ii.shape[1:])) for ii in m.inputs] ri = keras.layers.Input((sum(bs),))