need to better understand how to regularize the loss function with the generated rain rate
need to better understand how to regularize the loss function with the generated rain rate