numpy==1.26.4
ptflops==0.7.3
PuLP==2.8.0
PyYAML==6.0.1
timm==1.0.3
torch==2.3.0
torchvision==0.18.0
wandb==0.17.0
