Added in the training function to Samsara.