search

tensorflow / tpu

Reference models and tools for Cloud TPUs.
https://cloud.google.com/tpu/
Apache License 2.0
5.21k stars 1.77k forks source link

How to use tf.keras to train on jupyter notebook #834

Open LINYOUWEI0804 opened 4 years ago

LINYOUWEI0804 commented 4 years ago

There is no teaching with keras training on the Internet. Is the use of transfer learning like MobileNet?
ex. model = MobileNet(weights='imagenet',include_top=False,input_shape = (224,224,3))

My version: ubuntu 18.04 python 3.6 tf-nightly-gpu 2.4.0.dev20200811 Keras 2.3.1

Hope to provide training code. Thank!!

mikegre-google commented 4 years ago

I found this Colab that shows how to use tf.keras: Simple Classification Model using Keras on Colab TPU.ipynb.

I hope this helps!

swghosh commented 4 years ago

Why not: https://www.tensorflow.org/tutorials/images/transfer_learning?