diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-08-06 16:49:10 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-08-06 16:49:10 -0500 |
| commit | 3729c6d8467f7a0b384c176e203103c920993843 (patch) | |
| tree | 178265f526cf3d43457f6d512a8f0d11e37326b0 /experiments/rain_ep_bias_train.py | |
| parent | 2b3cc344d76fd1a73b5abc6aaac198ea02e6303d (diff) | |
feat: support frozen Rain neutral predictors
Diffstat (limited to 'experiments/rain_ep_bias_train.py')
| -rw-r--r-- | experiments/rain_ep_bias_train.py | 4 |
1 files changed, 3 insertions, 1 deletions
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) |