diff --git a/deeplift/conversion/kerasapi_conversion.py b/deeplift/conversion/kerasapi_conversion.py index 2b4b9f2..48c2eda 100644 --- a/deeplift/conversion/kerasapi_conversion.py +++ b/deeplift/conversion/kerasapi_conversion.py @@ -404,7 +404,8 @@ def insert_weights_into_nested_model_config(nested_model_weights, layer_name = layer_config["config"]["name"] layer_weights = [np.array(nested_model_weights[x]) for x in nested_model_weights.keys() if - x.startswith(layer_name+"/")] + (x.decode("utf-8") if hasattr(x, 'decode') + else x).startswith(layer_name+"/")] if (len(layer_weights) > 0): layer_config["config"]["weights"] = layer_weights