From f38216f43d1939976b440b71290c18ffaab278fa Mon Sep 17 00:00:00 2001
From: Yen-Chen Lin <yenchenlin1994@gmail.com>
Date: Fri, 24 Apr 2020 04:13:14 -0400
Subject: [PATCH] Fix iter

---
 run_nerf.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/run_nerf.py b/run_nerf.py
index 8291747..fab80da 100644
--- 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()