fix typo in negative sample selection

finetune.py

@@ -29,7 +29,7 @@ class ESCIDataset(Dataset):
                     for p in query['e']:
                         positive = p['title']
                         for n in query['i']:
-                            negative = p['title']
+                            negative = n['title']
                             self.queries.append(InputExample(texts=[query['query'], positive, negative]))
 
     def __getitem__(self, item):
@@ -68,11 +68,11 @@ model.fit(train_objectives=[(train_dataloader, train_loss)],
           epochs=num_epochs,
           warmup_steps=warmup_steps,
           use_amp=True,
-          checkpoint_path=model_save_path,
-          checkpoint_save_steps=len(train_dataloader),
+#          checkpoint_path=model_save_path,
+#          checkpoint_save_steps=len(train_dataloader),
           optimizer_params = {'lr': lr},
-#          evaluator=evaluator,
-#          evaluation_steps=300,
+          evaluator=evaluator,
+#          evaluation_steps=1000,
           output_path=model_save_path
           )
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5aae60cda92f5f51c6aa5d10b1cee0266982c0c10becef516cc9062d2bad9b67
+oid sha256:089d832e391c446f9bc7df2f9b5f772405a7e762506908f9a22b6a71fb172e74
 size 90891565

pytorch_model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:05218f129c6611d20167c617dc6f6df81a45fdbc8d094ba8cdd1bfbc96aeb947
+oid sha256:cfa7e76b1b76b1617d7188ce710c8f03ac0c85463f8ad55b276c484d2c7c725a
 size 90984263