diff options
Diffstat (limited to 'train_lora.py')
-rw-r--r-- | train_lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index 51dc827..1ff25ff 100644 --- a/train_lora.py +++ b/train_lora.py | |||
@@ -1137,7 +1137,7 @@ def main(): | |||
1137 | lora_datamodule = create_datamodule( | 1137 | lora_datamodule = create_datamodule( |
1138 | valid_set_size=args.valid_set_size, | 1138 | valid_set_size=args.valid_set_size, |
1139 | batch_size=args.train_batch_size, | 1139 | batch_size=args.train_batch_size, |
1140 | dropout=args.tag_dropout, | 1140 | tag_dropout=args.tag_dropout, |
1141 | filter=partial(keyword_filter, None, args.collection, args.exclude_collections), | 1141 | filter=partial(keyword_filter, None, args.collection, args.exclude_collections), |
1142 | ) | 1142 | ) |
1143 | lora_datamodule.setup() | 1143 | lora_datamodule.setup() |