diff --git a/run_nerf.py b/run_nerf.py index a3e1f37..8291747 100644 --- a/run_nerf.py +++ b/run_nerf.py @@ -804,7 +804,7 @@ def train(): - if i%args.i_print==0 or i < 10: + if i%args.i_print==0: tqdm.write(f"[TRAIN] Iter: {i} Loss: {loss.item()} PSNR: {psnr.item()}") """ print(expname, i, psnr.numpy(), loss.numpy(), global_step.numpy())