tensorflow / cloud

The TensorFlow Cloud 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.
https://github.com/tensorflow/cloud
Apache License 2.0
372 stars 85 forks source link

Create models module on TF Cloud experimental with run_models wrapper #343

Closed copybara-service[bot] closed 3 years ago

copybara-service[bot] commented 3 years ago

Create models module on TF Cloud experimental with run_models wrapper

Add wrapper for tfc.run that runs models from TF Model Garden on the Cloud with TFDS. This method allows for running models from TF Model Garden using datasets from TFDS directly on Google Cloud. Currently it only supports image classification models. Specifically ResNet and EfficientNet.