torch<2.3.0,>=2.0.0
PyYAML>=6.0
tqdm
numpy<2
matplotlib
seaborn
pandas
torchvision>=0.15.0
onnx
onnxruntime
onnxruntime-tools
timm

[dev]
pytest
pytest-cov
flake8
mypy
black
