From 3311b3c94ec265534b7ed51fe4c98ca5b99d97ba Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Wed, 20 Apr 2022 14:02:44 -0700 Subject: [PATCH] run black --- mattport/nerf/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mattport/nerf/trainer.py b/mattport/nerf/trainer.py index bb4e39d64..ea67e9b24 100644 --- a/mattport/nerf/trainer.py +++ b/mattport/nerf/trainer.py @@ -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])