Skip to content

Commit

Permalink
fix(executor): fix init
Browse files Browse the repository at this point in the history
Co-authored-by: Nan Wang <nan.wang@jina.ai>
  • Loading branch information
anish2197 and nan-wang committed Jun 27, 2020
1 parent 02e02dc commit 4fd2ff2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jina/executors/encoders/image/customtfkeras.py
Expand Up @@ -16,7 +16,7 @@ class CustomImageKerasEncoder(BaseCVTFEncoder):
https://www.tensorflow.org/api_docs/python/tf/keras/applications
"""

def __init__(self, model_path: str, layer_name: str, *args, **kwargs):
def __init__(self, model_path: str, layer_name: str, channel_axis: int = -1, *args, **kwargs):

"""
:param model_path: the path where the model is stored.
Expand Down

0 comments on commit 4fd2ff2

Please sign in to comment.