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 6c35d41..ef39c38 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -788,7 +788,7 @@ def main(): | |||
788 | args.num_vectors, | 788 | args.num_vectors, |
789 | args.train_data_template | 789 | args.train_data_template |
790 | ): | 790 | ): |
791 | run(i, [placeholder_token], [initializer_token], [num_vectors], data_template) | 791 | run(i, [placeholder_token], [initializer_token], num_vectors, data_template) |
792 | embeddings.persist() | 792 | embeddings.persist() |
793 | 793 | ||
794 | 794 | ||