set cuda as default

nerfstudio-project · Nov 24, 2024 · 5aba78e · 5aba78e
1 parent fe48ffa
commit 5aba78e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nerfstudio/configs/base_config.py b/nerfstudio/configs/base_config.py
@@ -68,7 +68,7 @@ class MachineConfig(PrintableConfig):
     """current machine's rank (for DDP)"""
     dist_url: str = "auto"
     """distributed connection point (for DDP)"""
-    device_type: Literal["cpu", "cuda", "mps"]
+    device_type: Literal["cpu", "cuda", "mps"] = "cuda"
     """device type to use for training"""