【benchmark】fix benchmark prepare.sh (#3803)

* fix profiler

* add max_mem_reserved for benchmark

* fix benchmark

* Update prepare.sh

* Update prepare.sh
pull/3807/head
gmm 6 months ago committed by GitHub
parent 91170bd260
commit 98fe6d1153
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -35,6 +35,8 @@ if [[ ${MODE} = "benchmark_train" ]];then
pip install setuptools_scm #-i https://pypi.tuna.tsinghua.edu.cn/simple pip install setuptools_scm #-i https://pypi.tuna.tsinghua.edu.cn/simple
pip install . #-i https://pypi.tuna.tsinghua.edu.cn/simple pip install . #-i https://pypi.tuna.tsinghua.edu.cn/simple
pip install jsonlines pip install jsonlines
pip install -U scipy==1.12.0 # 高版本数据处理部分报错
pip install -U matplotlib==3.7.1 # 高版本报错cannot import name 'get_cmap' from 'matplotlib.cm'
pip list pip list
cd - cd -
if [[ ${model_name} == "conformer" ]]; then if [[ ${model_name} == "conformer" ]]; then

Loading…
Cancel
Save