From 3729c6d8467f7a0b384c176e203103c920993843 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Thu, 6 Aug 2026 16:49:10 -0500 Subject: feat: support frozen Rain neutral predictors --- experiments/rain_ep_bias_train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'experiments') diff --git a/experiments/rain_ep_bias_train.py b/experiments/rain_ep_bias_train.py index 8b6e9ad..7bdff26 100644 --- a/experiments/rain_ep_bias_train.py +++ b/experiments/rain_ep_bias_train.py @@ -35,7 +35,9 @@ def parse_args() -> argparse.Namespace: "--mode", choices=sorted(RainGradientCorrector.MODES), required=True) parser.add_argument("--bias-ratio", type=float, default=0.5) parser.add_argument("--predictor-rate", type=float, default=0.1) - parser.add_argument("--neutral-cadence", type=int, default=1) + parser.add_argument( + "--neutral-cadence", type=int, default=1, + help="training steps per neutral update; zero freezes after calibration") parser.add_argument("--calibration-batches", type=int, default=0) parser.add_argument("--epochs", type=int, default=2) parser.add_argument("--train-limit", type=int, default=2048) -- cgit v1.2.3