This repository provides APIs that will allow to easily go from debugging and training your Keras and TensorFlow code in a local environment to distributed training in the cloud.
- Python >= 3.5
- Set up your Google Cloud project
- Authenticate your GCP account
- nbconvert - if you are using an iPython notebook
pip install -U tensorflow-cloud
git clone https://github.com/tensorflow/cloud.git
cd cloud
pip install .
- Usage with
tf.kerasscript that trains usingmodel.fit. - Usage with
tf.kerasscript that trains using a custom training loop.
We welcome community contributions, see CONTRIBUTING.md and, for style help, Writing TensorFlow documentation guide.