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 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,