INNER CODE UNIT · Python
max_epoch
junshutang/Make-It-3D · main.py:235
max_epoch = np.ceil(opt.iters / 100).astype(np.int32)
trainer.train(train_loader, valid_loader, max_epoch)
# also test
if opt.final:
test_loader = NeRFDataset(opt, device=device, type='test', H=opt.H, W=opt.W, size=64).dataloader()
trainer.test(test_loader, write_image=False, write_video=True)
if opt.save_mesh:
trainer.save_mesh(resolution=256)
if opt.refine:
mv_loader = NeRFDataset(opt, device=device, type='gen_mv', H=opt.H, W=opt.W, size=33).dataloader()
test_loader = NeRFDataset(opt, device=device, type='test', H=opt.H, W=opt.W, size=64).dataloader()
trainer.test(mv_loader, save_path=os.path.join(opt.workspace, 'mvimg'), write_image=True, write_video=False)
trainer.refine(os.path.join(opt.workspace, 'mvimg'), opt.refine_iters, test_loader)