Fix iter

4 years ago · f38216f43d
parent 649045ea60
commit f38216f43d
1 changed files with 3 additions and 2 deletions
--- a/run_nerf.py
+++ b/run_nerf.py
@ -224,7 +224,7 @@ def create_nerf(args):
        print('Reloading from', ckpt_path)
        ckpt = torch.load(ckpt_path)

-        start = ckpt['global_step'] + 1
+        start = ckpt['global_step']
        optimizer.load_state_dict(ckpt['optimizer_state_dict'])

        # Load model
@ -687,7 +687,8 @@ def train():

    # Summary writers
    # writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
-
+    
+    start = start + 1
    for i in trange(start, N_iters):
        time0 = time.time()