Skip to content

Conversation

@graykode
Copy link

@tushuhei tushuhei self-requested a review August 19, 2019 02:43
Copy link
Member

@tushuhei tushuhei left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for your PR! Please update your code to work with the new version of retrain.py. Also, it'll be great if you could check if your refined code works with your training dataset by running retraining.

Thanks,
Shuhei

RUN git clone --depth 1 https://github.com/tensorflow/hub --branch r0.5

CMD python3 /tensorflow/tensorflow/examples/image_retraining/retrain.py \
CMD python3 /hub/examples/image_retraining/retrain.py \
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The new retrain.py script does not support architecture flag, so we need to specify the model architecture using tfhub_module flag instead. Could you update the flag accordingly?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants