run black

This commit is contained in:
Matt Tancik committed 2022-04-20 14:02:44 -07:00
1 parent 40f0338eb2
commit 3311b3c94e
1 file changed
+1 -1
+1 -1
View File
@@ -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])