We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fe48ffa commit 5aba78eCopy full SHA for 5aba78e
nerfstudio/configs/base_config.py
@@ -68,7 +68,7 @@ class MachineConfig(PrintableConfig):
68
"""current machine's rank (for DDP)"""
69
dist_url: str = "auto"
70
"""distributed connection point (for DDP)"""
71
- device_type: Literal["cpu", "cuda", "mps"]
+ device_type: Literal["cpu", "cuda", "mps"] = "cuda"
72
"""device type to use for training"""
73
74
0 commit comments