matplotlib.pyplot as plt
numpy
datasets
transformers
re
torch
torch.nn.functional
tqdm
random
sklearn.metrics 
roc_curve, precision_recall_curve, auc
argparse
datetime
os
json
functools
custom_datasets
multiprocessing.pool 
ThreadPool
time