undoing changes in main_tensoRF.py

lishy66 · Feb 15, 2023 · a721ffd · a721ffd
1 parent bfe87fe
commit a721ffd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main_tensoRF.py b/main_tensoRF.py
@@ -127,6 +127,7 @@
  scheduler = lambda optimizer: optim.lr_scheduler.LambdaLR(optimizer, lambda iter: 0.1 ** min(iter / opt.iters, 1))
 
  trainer = Trainer('ngp', opt, model, device=device, workspace=opt.workspace, optimizer=optimizer, criterion=criterion, ema_decay=None, fp16=opt.fp16, lr_scheduler=scheduler, scheduler_update_every_step=True, metrics=[PSNRMeter()], use_checkpoint=opt.ckpt, eval_interval=50)
+
  # calc upsample target resolutions
  upsample_resolutions = (np.round(np.exp(np.linspace(np.log(opt.resolution0), np.log(opt.resolution1), len(opt.upsample_model_steps) + 1)))).astype(np.int32).tolist()[1:]
  print('upsample_resolutions:', upsample_resolutions)