saved model + onnx

laitassou · Jun 12, 2020 · 2427dc3 · 2427dc3
1 parent 4b5070b
commit 2427dc3
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 1 deletion.
diff --git a/models_mpg/saved_model.pb b/models_mpg/saved_model.pb
diff --git a/models_mpg/variables/variables.data-00000-of-00001 b/models_mpg/variables/variables.data-00000-of-00001
diff --git a/models_mpg/variables/variables.index b/models_mpg/variables/variables.index
diff --git a/mpg.onnx b/mpg.onnx
diff --git a/regression/fuel.py → regression/mpg.py b/regression/fuel.py → regression/mpg.py
@@ -151,7 +151,7 @@ def build_model():
 
 
 # Save model to SavedModel format
-tf.saved_model.save(model, "./models")
+tf.saved_model.save(model, "./models_mpg")
 
 
 # Convert Keras model to ConcreteFunction