Skip to content

Commit 5aba78e

Browse files
author
Noé Pion
committed
set cuda as default
1 parent fe48ffa commit 5aba78e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nerfstudio/configs/base_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ class MachineConfig(PrintableConfig):
6868
"""current machine's rank (for DDP)"""
6969
dist_url: str = "auto"
7070
"""distributed connection point (for DDP)"""
71-
device_type: Literal["cpu", "cuda", "mps"]
71+
device_type: Literal["cpu", "cuda", "mps"] = "cuda"
7272
"""device type to use for training"""
7373

7474

0 commit comments

Comments
 (0)