added rmse_loss layer