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 902f508..7900fbd 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -738,7 +738,7 @@ def main(): | |||
738 | 738 | ||
739 | stats = list(zip(placeholder_tokens, placeholder_token_ids, initializer_tokens, initializer_token_ids)) | 739 | stats = list(zip(placeholder_tokens, placeholder_token_ids, initializer_tokens, initializer_token_ids)) |
740 | 740 | ||
741 | print(f"{i + 1}: {stats})") | 741 | print(f"{i + 1}: {stats}") |
742 | 742 | ||
743 | datamodule = VlpnDataModule( | 743 | datamodule = VlpnDataModule( |
744 | data_file=args.train_data_file, | 744 | data_file=args.train_data_file, |