Latency_Prediction_by_nn-Meter / requirements.txt
Jiahang's picture
update script
aefff9b
-e git+https://github.com/microsoft/nn-Meter.git@dev/block-latency#egg=nn-Meter
tensorflow==2.6.0
torch==1.9.0
torchvision==0.10.0
onnx==1.10.0
onnx-simplifier==0.3.6