Fix

1 files changed, 1 insertions, 1 deletions

diff --git a/train_ti.py b/train_ti.py
index 2d51800..ca5b113 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -811,7 +811,6 @@ def main():
         print(f"============ TI batch {i + 1} ============")
         print("")
         print(stats)
-        print("")
 
         filter_tokens = [token for token in args.filter_tokens if token in placeholder_tokens]
 
@@ -874,6 +873,7 @@ def main():
         training_iter = 1
 
         while continue_training:
+            print("")
             print(f"------------ TI cycle {training_iter} ------------")
             print("")