The Configuration file
The ELBO configuration is specified in YAML in a configuration file. Let's look at its contents.
The configuration file, typically named elbo.yaml
has the following properties:
Option | Description | Examples |
---|---|---|
| The name of your ML training task | "Hello, ELBO 💪" |
| The class of GPU you want to request. This can be one of the following:
|
|
| A setup script that will be run prior to calling the training code |
|
| A | |
| The main training code. The task execution will call this file directly. |
|
| The directory where this task is present. Usually the current directory. This directory will be zipped and uploaded for running the task. Please make sure all the files and scripts needed to run the training code are present in this directory. |
|
| The directory where your code will place model checkpoints, plots, generated files etc. The ELBO service will package this directory and save it for you to download after the task is complete. |
|
| Setting this to |
|
Tip: If you are submitting the task for the first time, you may want to run the training task on an Economy
class machine and then move to higher classes when you see the model converging after a few epochs.
Here is a sample configuration with comments on what each property means:
Last updated