Skip to content

Commit c9026b4

Browse files
committed
fix a bug in AbsDataset.py for embedder training
1 parent d6409b8 commit c9026b4

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

FlagEmbedding/abc/finetune/embedder/AbsDataset.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -426,6 +426,8 @@ def _get_train_group_size(self, batch_raw_data):
426426
return 2, data_type
427427
elif data_type in ['symmetric_class']:
428428
return min(len(batch_raw_data['neg'][0]) + 1, self.args.train_group_size), data_type
429+
else:
430+
return self.args.train_group_size, data_type
429431
return self.args.train_group_size, None
430432

431433
def _create_batch_data(self, batch_raw_data):

0 commit comments

Comments
 (0)