batch_size: 64
epochs: 5