Skip to content

Commit 9d08f69

Browse files
authored
Update train_sts_qqp_crossdomain.py
Fix #1014
1 parent c04c55b commit 9d08f69

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/training/data_augmentation/train_sts_qqp_crossdomain.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
python train_sts_qqp_crossdomain.py pretrained_transformer_model_name
1818
"""
1919
from torch.utils.data import DataLoader
20-
from sentence_transformers import models, losses, util, LoggingHandler
20+
from sentence_transformers import models, losses, util, LoggingHandler, SentenceTransformer
2121
from sentence_transformers.cross_encoder import CrossEncoder
2222
from sentence_transformers.cross_encoder.evaluation import CECorrelationEvaluator
2323
from sentence_transformers.evaluation import EmbeddingSimilarityEvaluator, BinaryClassificationEvaluator

0 commit comments

Comments
 (0)