loss : mse metric : mae lr : 0.0001 num_epochs : 100