Skip to content

T
tic
Commit 492a37c5 by Yuwei Hu Committed by Tianqi Chen
Options

[FIX] make keras frontend py3 compatible (#283)

parent 999dd1ef
Showing with 2 additions and 2 deletions
nnvm/python/nnvm/frontend/keras.py
...@@ -17,7 +17,7 @@ def _check_data_format(keras_layer): ...@@ -17,7 +17,7 @@ def _check_data_format(keras_layer):
def _convert_activation(insym, keras_layer, _): def _convert_activation(insym, keras_layer, _):
if isinstance(keras_layer, (str, unicode)): if isinstance(keras_layer, str):
act_type = keras_layer act_type = keras_layer
else: else:
if sys.version_info.major < 3: if sys.version_info.major < 3:
...@@ -25,7 +25,7 @@ def _convert_activation(insym, keras_layer, _): ...@@ -25,7 +25,7 @@ def _convert_activation(insym, keras_layer, _):
else: else:
act_type = keras_layer.activation.__name__ act_type = keras_layer.activation.__name__
if act_type == 'linear': if act_type == 'linear':
if isinstance(keras_layer, (str, unicode)): if isinstance(keras_layer, str):
return insym return insym
return _sym.__add_scalar__(_sym.__mul_scalar__(insym, \ return _sym.__add_scalar__(_sym.__mul_scalar__(insym, \
scalar=keras_layer.alpha), scalar=keras_layer.beta) scalar=keras_layer.alpha), scalar=keras_layer.beta)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment