rm redundant print
This commit is contained in:
parent
35a2e719d2
commit
356125d30f
1 changed files with 1 additions and 1 deletions
|
@ -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()}")
|
tqdm.write(f"[TRAIN] Iter: {i} Loss: {loss.item()} PSNR: {psnr.item()}")
|
||||||
"""
|
"""
|
||||||
print(expname, i, psnr.numpy(), loss.numpy(), global_step.numpy())
|
print(expname, i, psnr.numpy(), loss.numpy(), global_step.numpy())
|
||||||
|
|
Loading…
Reference in a new issue