From 2036736c9e97058635540f3aa790fbc8cc228fe6 Mon Sep 17 00:00:00 2001
From: dkumazaw <dkumazaw@stanford.edu>
Date: Wed, 22 May 2019 15:38:03 +0900
Subject: [PATCH] Fix function call

---
 runtime-modeling/main_tflite.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/runtime-modeling/main_tflite.py b/runtime-modeling/main_tflite.py
index 7dd930e..e1c210f 100644
--- 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,