Train

Training machine learning models often requires custom train loop and custom code. As such, we don’t provide an out of the box training loop app. We do however have examples for how you can construct your training app as well as generic components you can use to run your custom training app.

Trainer Examples

• Trainer Component Example

• Trainer App Example

Components

These are generic components for common patterns that that can be used in your train components.

• Distributed Data Parallel Component - torchx.components.dist.ddp()