ENOT-AutoDL
/

imagenet-benchmark

Image Classification

Inference Endpoints

Model card Files Files and versions Community

imagenet-benchmark / requirements.txt

savchenkoyana's picture

add ViT x4.8 ONNX, small fixes in test.py, and allow measuring macs on ONNX

061bac4 over 1 year ago

115 Bytes

	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