otthorn/nerf-pytorch
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Fix iter

Browse source
This commit is contained in:
Yen-Chen Lin 2020-04-24 04:13:14 -04:00
parent 649045ea60
commit f38216f43d
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
run_nerf.py
View file

@ -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():
# 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()