train doc

MaxThFe · Sep 1, 2022 · 7b528b8 · 7b528b8
1 parent c22db87
commit 7b528b8
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/data/__pycache__/cycles.cpython-310.pyc b/data/__pycache__/cycles.cpython-310.pyc
diff --git a/decoder/__pycache__/utils.cpython-310.pyc b/decoder/__pycache__/utils.cpython-310.pyc
diff --git a/main.py b/main.py
@@ -2,7 +2,7 @@
 import os
 import datetime
 import time
-import tqdm
+from tqdm import tqdm
 
 import torch
 from torch.optim import Adam
@@ -60,6 +60,7 @@ def main(opts):
             batch_count += 1
 
             if batch_count % opts['batch_size'] == 0:
+                print('\n')
                 printer.update(epoch + 1, {'averaged_loss': batch_loss/opts['batch_size']})
 
                 if opts['clip_grad']: