loss : mse
metric : mae
lr : 0.001
num_epochs : 10