rm redundant print

4 years ago · 356125d30f
parent 35a2e719d2
commit 356125d30f
1 changed files with 1 additions and 1 deletions
--- 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())