numpy == 1.21.6
pyyaml == 6.0
timm == 0.5.4
einops == 0.4.1
onnx == 1.12.0
onnx-simplifier == 0.4.7
torch
torchvision
onnxruntime