From eb2416ee22f916f4e1d29405d634a0ee613a525c Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Thu, 6 Aug 2026 16:50:51 -0500 Subject: fix: mark Rain determinism as best effort --- experiments/rain_ep_bias_train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'experiments/rain_ep_bias_train.py') 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, -- cgit v1.2.3