https://docs.viam.com/ml/train-model/