diff options
Diffstat (limited to 'train_ti.py')
| -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, |