Skip to main content

Training models


You need to create a model before training it.

Video Tutorial - Building micro-models

Training a model

  1. The first screen on the model training dashboard i.e. ‘Training settings’ displays the model type, the relevant ontology objects and allows users to define:
  • Number of epochs
  • Advanced settings:
    • Processing unit: CPU or GPU
    • Batch size
    • Training weights: pre-trained weights to use; these vary according to the selected framework and model architecture
  1. After making the appropriate choices, the second screen offers the option to select which data units to use for training. In the example above, we only have two image groups to select from and only selected the reviewed image group for training.

Once you're ready, press Train to start the process!

Monitor training progress

Once a training job is launched you can monitor its progress via the     icon in the upper right navigation bar. While the model is training, you will be updated with the current epoch number and loss value.

Once the training has finished, a     will appear along with an indication of how long ago the training job completed.

Congratulations! You've successfully trained a model, proceed to our section on Inference to learn how to apply your model to generate automated labels.

Model training logs

Once model training is finished, we can explore the training logs through the relevant option under the model tab. From here we can download the training log or visualise the loss curve in the UI.