Commit 356125d3 by Yen-Chen Lin

rm redundant print

parent 35a2e719
......@@ -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())
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment