summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--train_ti.py2
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():
811 print(f"============ TI batch {i + 1} ============") 811 print(f"============ TI batch {i + 1} ============")
812 print("") 812 print("")
813 print(stats) 813 print(stats)
814 print("")
815 814
816 filter_tokens = [token for token in args.filter_tokens if token in placeholder_tokens] 815 filter_tokens = [token for token in args.filter_tokens if token in placeholder_tokens]
817 816
@@ -874,6 +873,7 @@ def main():
874 training_iter = 1 873 training_iter = 1
875 874
876 while continue_training: 875 while continue_training:
876 print("")
877 print(f"------------ TI cycle {training_iter} ------------") 877 print(f"------------ TI cycle {training_iter} ------------")
878 print("") 878 print("")
879 879