Fix minor bug 'force_compute_all' not a vaild argument

DigitalPhonetics · Dec 19, 2023 · 98362ae · 98362ae
1 parent a7439c5
commit 98362ae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/run_anonymization.py b/run_anonymization.py
@@ -28,6 +28,6 @@
     else:
         devices.append(torch.device('cpu'))
 
-    pipeline = PIPELINES[config['pipeline']](config=config, force_compute_all=args.force_compute, devices=devices)
-    pipeline.run_anonymization_pipeline(datasets)
-
+    with torch.no_grad():
+        pipeline = PIPELINES[config['pipeline']](config=config, force_compute=args.force_compute, devices=devices)
+        pipeline.run_anonymization_pipeline(datasets)