diff options
| -rw-r--r-- | train_ti.py | 2 |
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 | ||
