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