which loss function te be used ? i used mse loss function. How much epochs required for this please. suggest me