diff --git a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py index 87a55913d997c6a893b5075ace6d07f988db0c29..8f39d77987fe0a4b595f48276480b80187047ea2 100644 --- a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py +++ b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py @@ -459,7 +459,7 @@ class TensorRtTranslator: if strides[0] > 1 and cur_node.padding.strip() == "same": print ("!ERROR: Same Padding not supported for Conv with Stride > 1") - print ("Use: ZeroPadding2D(padding=(" + str(padding) + "," + str(padding) + "));\n"); + print ("Use: ZeroPadding2D(padding=(" + str(padding) + "," + str(padding) + ")) before the Conv2D/DepthwiseConv2D Operator \n"); sys.exit(0) # NOTE: For Json (tuning config) file generation @@ -467,8 +467,6 @@ class TensorRtTranslator: self.addConvOverheads(weights, padding, strides) elif layer_type == "DepthwiseConv2D": - #self.json_str += "depthwise_convolution_" + str(self.op_count) + " : 0, \n" - #self.op_count += 1 self.addBaselineKnob("depthwise_convolution") @@ -1148,7 +1146,7 @@ def translate_to_approxhpvm(model, if reload_weights: print ("NOTE: Using existing pretrained weights \n") else: - print ("NOTE: dumping new set of weights \n") + print ("NOTE: Translating Keras .h5 file to HPVM .bin files \n") print ("-- Weight Files Under : ", weights_dir) print ("-- TensorRT src : ", src_dir + "/src.cc")