Skip to content

ymwangg/bench-bert

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
README.md
README.md
 
 
bench_onnx.py
bench_onnx.py
 
 
bench_pt.py
bench_pt.py
 
 
bench_tvm.py
bench_tvm.py
 
 
bench_tvm_multiprocess.py
bench_tvm_multiprocess.py
 
 
compile_tvm_model.py
compile_tvm_model.py
 
 
convert_graph_to_onnx.py
convert_graph_to_onnx.py
 
 
download_all_onnx_models.sh
download_all_onnx_models.sh
 
 
download_all_pt_models.sh
download_all_pt_models.sh
 
 
download_onnx_model.sh
download_onnx_model.sh
 
 
download_pt_model.py
download_pt_model.py
 
 
download_pt_model.sh
download_pt_model.sh
 
 
evaluate.sh
evaluate.sh
 
 
models.txt
models.txt
 
 
profile_cublas_trt.py
profile_cublas_trt.py
 
 
profile_onnx_tvm.py
profile_onnx_tvm.py
 
 
profile_pt.sh
profile_pt.sh
 
 
profile_pt_tvm.py
profile_pt_tvm.py
 
 
pt-tvm.ipynb
pt-tvm.ipynb
 
 
run_onnx.py
run_onnx.py
 
 
run_onnx.sh
run_onnx.sh
 
 
run_pt.py
run_pt.py
 
 
run_tvm.py
run_tvm.py
 
 

Repository files navigation

Prerequisites

pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install transformers==4.6.1

About

No description, website, or topics provided.

Resources

Readme
Activity

Stars

2 stars

Watchers

2 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages