timm==0.5.4
tensorboardX
numpy