diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-08-06 16:50:51 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-08-06 16:50:51 -0500 |
| commit | eb2416ee22f916f4e1d29405d634a0ee613a525c (patch) | |
| tree | 65d80143f9d185032f0799d9527890c435b319aa /experiments/rain_ep_bias_train.py | |
| parent | 3729c6d8467f7a0b384c176e203103c920993843 (diff) | |
fix: mark Rain determinism as best effort
Diffstat (limited to 'experiments/rain_ep_bias_train.py')
| -rw-r--r-- | experiments/rain_ep_bias_train.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/experiments/rain_ep_bias_train.py b/experiments/rain_ep_bias_train.py index 7bdff26..defd973 100644 --- a/experiments/rain_ep_bias_train.py +++ b/experiments/rain_ep_bias_train.py @@ -101,7 +101,7 @@ def main() -> None: if args.deterministic: torch.backends.cudnn.benchmark = False torch.backends.cudnn.deterministic = True - torch.use_deterministic_algorithms(True) + torch.use_deterministic_algorithms(True, warn_only=True) device = torch.device(args.device) training_data, test_data = load_fashion_mnist( @@ -242,7 +242,8 @@ def main() -> None: "inference_iterations": args.inference_iterations, "seed": args.seed, "device": str(device), - "deterministic": args.deterministic, + "determinism": ( + "best_effort_warn_only" if args.deterministic else "author_default"), "autodiff_used_for_learning": False, }, "metrics": metrics, |