imagenet-benchmark / requirements.txt
savchenkoyana's picture
add ViT x4.8 ONNX, small fixes in test.py, and allow measuring macs on ONNX
061bac4
torch==1.13.1
torchvision==0.14.1
fvcore==0.1.5.post20221221
onnxruntime-gpu==1.15.1
onnx==1.13.1
onnx2torch==1.5.6