Commit f38216f4 by Yen-Chen Lin

Fix iter

parent 649045ea
...@@ -224,7 +224,7 @@ def create_nerf(args): ...@@ -224,7 +224,7 @@ def create_nerf(args):
print('Reloading from', ckpt_path) print('Reloading from', ckpt_path)
ckpt = torch.load(ckpt_path) ckpt = torch.load(ckpt_path)
start = ckpt['global_step'] + 1 start = ckpt['global_step']
optimizer.load_state_dict(ckpt['optimizer_state_dict']) optimizer.load_state_dict(ckpt['optimizer_state_dict'])
# Load model # Load model
...@@ -687,7 +687,8 @@ def train(): ...@@ -687,7 +687,8 @@ def train():
# Summary writers # Summary writers
# writer = SummaryWriter(os.path.join(basedir, 'summaries', expname)) # writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
start = start + 1
for i in trange(start, N_iters): for i in trange(start, N_iters):
time0 = time.time() time0 = time.time()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment