torch transformers #pip install -U sentence-transformers import numpy as np import sys import json from torch.utils.data import DataLoader from sentence_transformers import SentenceTransformer, LoggingHandler, util, models, evaluation, losses, InputExample import logging from datetime import datetime import gzip import os import tarfile from collections import defaultdict from torch.utils.data import IterableDataset import tqdm from torch.utils.data import Dataset