googlehydrology.training.loss module