turn optimizations back on

mlcommons · Sep 22, 2023 · 97d1768 · 97d1768
1 parent bf47c47
commit 97d1768
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reference_algorithms/target_setting_algorithms/pytorch_nadamw.py b/reference_algorithms/target_setting_algorithms/pytorch_nadamw.py
@@ -58,7 +58,7 @@ def __init__(self,
       raise ValueError(f'Invalid weight_decay value: {weight_decay}')
     defaults = {
         'lr': lr, 'betas': betas, 'eps': eps, 'weight_decay': weight_decay,
-        'foreach': False, 'fused': False,
+        # 'foreach': False, 'fused': False,
     }
     super().__init__(params, defaults)