Model Definitions

The model definition is the interface between Determined and the user’s application framework (e.g., Keras, TensorFlow), in terms of loading training data, describing a model architecture, and specifying the underlying iterative optimization training algorithms.

There interface you can use to write model definition is the Trial API. With the Trial API, developers expose the components of a model definition to Determined’s training loop implementation.

The command line interface and Python SDK support the Trial API.

To read more about the Trial API, see:

• Trial API

• determined.pytorch.PyTorchTrial

• determined.keras.TFKerasTrial

• determined.estimator.EstimatorTrial

Best Practices

To learn more about some best practices when writing Trial API model definitions, see Best Practices for Model Definitions.