EPOCHS: 300
BATCH_SIZE: 32
LEARNING_RATE: 0.001
TIME_STEPS: 60