fix compile warning

rafcabezas · Apr 18, 2020 · 6a8e87e · 6a8e87e
1 parent 40498ca
commit 6a8e87e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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),))