training:
    epochs: 2
    batch_size: 32
    learning_rate: 0.00005