Update main_tflite.py

enyac-group#17 : build_model is not defined in runtime-modeling/models.py
jsrimr · Dec 14, 2020 · 4ba35cb · 4ba35cb
1 parent 21c1f3e
commit 4ba35cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime-modeling/main_tflite.py b/runtime-modeling/main_tflite.py
@@ -308,7 +308,7 @@ def gen_model_fn(features, labels, mode, params):
   if FLAGS.min_depth:
     override_params['min_depth'] = FLAGS.min_depth
 
-  logits, _ = models.build_model(
+  logits, _ = models.build_mnasnet_model(
       features,
       model_name=FLAGS.model_name,
       training=is_training,