made float32 default

april-tools · Jul 21, 2024 · 0a1a1c5 · 0a1a1c5
1 parent 26c246b
commit 0a1a1c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/scripts/experiment.py b/src/scripts/experiment.py
@@ -224,7 +224,7 @@
 )
 parser.add_argument(
     "--dtype",
-    default="float64",
+    default="float32",
     choices=["float32", "float64"],
     help="The default Torch dtype to use",
 )

diff --git a/src/scripts/utils.py b/src/scripts/utils.py
@@ -111,6 +111,7 @@ def set_global_seed(seed: int, is_deterministic=True):
         if is_deterministic is True:
             torch.backends.cudnn.deterministic = True
             torch.backends.cudnn.benchmark = False
+    torch._dynamo.config.cache_size_limit = 24
 
 
 def get_git_revision_hash() -> str: