diff options
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, |