Train models¶

You can train your model in three simple steps:

Step 1. Initiate a new training.

Step 2. (Optional) Edit and analyze your training data.

Step 3. (Optional) Edit the parameters.

If the default training setup (auto setup) looks good to you, you don't have to update your training data or edit the training and model parameters. In this case, you can start the training immediately.

Step 1. Initiate a new training¶

1. Go to the Projects module.
2. Click the name of the necessary project.

3. In the Training center section, click Set up training.

   

   Note

   If you can't start a new training because the necessary button isn't available, make sure you have added annotated data to your project.

4. If necessary, change the training name.

   The training name is pre-filled with the project name.

5. Under Settings, select one of the following options:

   • Auto setup – to go with the default settings and start training immediately. If you select this option, you can click Train model and your training will start.
   • Advanced setup – to edit your training data or fine-tune training and model parameters.

   

Step 2. Edit and analyze your training data¶

You can edit and analyze your training data only if the Advanced setup option is selected.

1. In the Datasets section, if available, add the train and validation sets. For this, next to Training data, click Add.

   

2. In the sidebar, select the necessary datasets.

   If no models have been trained in the project or datasets created in the label center, the list of datasets will be empty. The following steps describe the flow with no datasets.

3. If there are no datasets in the list, do the following:

   1. Click Create new dataset.

      You are redirected to the label center where you can create a new dataset.

   2. Select the samples to add to a dataset. These can be:

      • All samples.
      • Samples that match specific filters (a class, a tag, and so on). If the result includes more than 100 samples, click Select all to select all filtered samples.
      • Samples that you select manually. However, you cannot select samples located on different pages. If you select samples, and then go to another page, these samples will no longer be selected.

   3. In the Define ground truth field, select what to include in your dataset—annotations by a specific member, last updated, or random annotations.

      While one sample can have many annotations, only one of the annotations can be used for training—the ground truth.

   4. Using the slider, define the train/validation split.

   5. Click Create & use for training.

      

      You will be redirected to the training setup, and your datasets will be added as the training data.

Step 3. Edit the parameters¶

1. On the left, in the Page content list, select the parameters that you want to edit.

2. In the corresponding section, edit the necessary parameters.

   For more information about the training parameters, hover over the info icon next to the necessary parameter.

3. To select a model for transfer learning:

   1. In the Transfer learning section, next to Model, click Add.
   2. Select a model whose weights (what it learned) you want to use for your new model.
   3. Click Add model.

4. Once your training is set up, click Train model.

What's next?¶

• If necessary, stop the training.
• Delete the failed or unnecessary training.
• Test your model.