Skip to content

Update image_retraining colab#609

Closed
djdongjin wants to merge 2 commits intotensorflow:masterfrom
djdongjin:make-image-classifier-colab
Closed

Update image_retraining colab#609
djdongjin wants to merge 2 commits intotensorflow:masterfrom
djdongjin:make-image-classifier-colab

Conversation

@djdongjin
Copy link
Contributor

This PR updates the image_retraining colab by replacing ImageDataGenerator with tf.data.Dataset pipeline and tf.image-based augmentation.

I will submit another PR that updates the make_image_classifier tool by also adding multi-GPU training.

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

Review Jupyter notebook visual diffs & provide feedback on notebooks.


Powered by ReviewNB

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

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants