From bddded403e8b9d45bb9efa9ef8e58591a4a9f5f6 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Thu, 11 Mar 2021 18:24:39 -0600
Subject: [PATCH] Keras frontend fix for MobileNet
---
hpvm/projects/keras/frontend/promise_translator.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/hpvm/projects/keras/frontend/promise_translator.py b/hpvm/projects/keras/frontend/promise_translator.py
index f655174c74..ee33b1c18a 100644
--- a/hpvm/projects/keras/frontend/promise_translator.py
+++ b/hpvm/projects/keras/frontend/promise_translator.py
@@ -306,7 +306,7 @@ class PromiseRtTranslator:
layer_str += op.activation_type + " fp32 1 "
if op_type == "DepthwiseConv2D":
- layer_str += "depthwise_conv fp32 1"
+ layer_str += "group_conv fp32 1"
if op.use_bias:
layer_str += "add "
if op.activation_type != "linear":
@@ -326,7 +326,7 @@ class PromiseRtTranslator:
if op_type == "MaxPooling2D":
layer_str += "pool_max fp32 1 "
- if op_type == "AvgPooling2D":
+ if op_type == "AveragePooling2D":
layer_str += "pool_mean fp32 1 "
if op_type == "Add":
@@ -570,6 +570,9 @@ class PromiseRtTranslator:
first_op = state.getFirstOp()
last_op = state.getFirstOp()
+ input_var = self.getSingleInputName(first_op)
+ output_var = self.getVariableName(last_op)
+
self.appendLayerString("BatchNorm", state)
state.clear()
@@ -637,8 +640,7 @@ class PromiseRtTranslator:
# FIXME: Only supporting single AveragePooling layers
def genPoolLayer(self, state):
- print ("\n\n Layer = ", state.op_string, "\n\n")
-
+
# For single pool layer should be all same
pool_op = state.getPoolOp()
--
GitLab