torch==1.7.0
torchvision==0.8.1
tqdm
pandas
func_timeout
wandb