From 6e889b242638b9e5613e98f7efa24b2eccfc9aee Mon Sep 17 00:00:00 2001 From: Kai Zhang Date: Sat, 17 Oct 2020 09:31:52 -0400 Subject: [PATCH] remove duplicate line --- ddp_train_nerf.py | 1 - 1 file changed, 1 deletion(-) diff --git a/ddp_train_nerf.py b/ddp_train_nerf.py index e6c5b94..80bfd76 100644 --- a/ddp_train_nerf.py +++ b/ddp_train_nerf.py @@ -323,7 +323,6 @@ def create_nerf(rank, args): for m in range(models['cascade_level']): for name in ['net_{}'.format(m), 'optim_{}'.format(m)]: models[name].load_state_dict(to_load[name]) - models[name].load_state_dict(to_load[name]) return start, models