torch==1.7.0
numpy==1.18.5
tqdm==4.46.1
thop==0.0.31.post2005241907
tensorboard
