Keras

+ Gradient

How to use Gradient and

Keras

together

Use Keras to train models on Gradient

Gradient supports any version of TensorFlow for Notebooks, Experiments, or Jobs. In Gradient, the ML framework used to execute workloads runs within a Docker container. Containers are lightweight and portable environments that may be customized to include various framework versions and other libraries.

Any Docker container is supported on the Gradient platform. This flexibility makes it easy to switch between different frameworks, to update containers from one version to another, and to incorporate other libraries to be used alongside the framework itself.

A set of pre-built containers is provided out of the box though any container can be used if hosted on a public or private container registry.

Running workloads with Keras

When launching a workload via the web interface, CLI, or automatically via a pipeline step, you can simply pass-in a Docker image path (e.g. <inline-code>ermaker/keras-jupyter<inline-code>. There are also several pre-configured templates available. These templates are updated regularly and optimized to run on Gradient.

A set of pre-built containers can be used as a starting point within Gradient

When using the CLI, the command would like something like this:

gradient notebooks create \  

--name "my job" \
--container ermaker/keras-jupyter \
--machineType P5000 \
--command "/paperspace/run.sh"
--projectId your-project-id
...