fix: disabled jax64 for evaluation

automl · Jun 4, 2024 · 7ce21f6 · 7ce21f6
1 parent fc094b7
commit 7ce21f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/configs/evaluate.yaml b/examples/configs/evaluate.yaml
@@ -12,7 +12,7 @@ hydra:
   job:
     chdir: true
 
-jax_enable_x64: true
+jax_enable_x64: false
 load_checkpoint: ""
 hpo_method: "default_method"