training: epochs: 3 batch_size: 16 learning_rate: 0.00005