numpy
tqdm
torch>1.6
torch4keras==0.2.9.post2
six