mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 03:14:00 +08:00
run black
This commit is contained in:
1 parent
40f0338eb2
commit
3311b3c94e
1 file changed
+1
-1
@@ -60,7 +60,7 @@ class Trainer:
|
||||
).to(f"cuda:{self.local_rank}")
|
||||
|
||||
## TODO(): we need to call load_checkpoint here if need be (before we wrap it in DDP)
|
||||
|
||||
|
||||
if self.world_size > 1:
|
||||
self.graph = DDP(self.graph, device_ids=[self.local_rank])
|
||||
dist.barrier(device_ids=[self.local_rank])
|
||||
|
||||
Reference in new issue
Block a user