From 98fe6d1153e59cbaf9653e00695e061169ec683c Mon Sep 17 00:00:00 2001
From: gmm <38800877+mmglove@users.noreply.github.com>
Date: Wed, 19 Jun 2024 17:36:59 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90benchmark=E3=80=91fix=20benchmark=20pr?=
 =?UTF-8?q?epare.sh=20(#3803)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* fix profiler

* add max_mem_reserved for benchmark

* fix benchmark

* Update prepare.sh

* Update prepare.sh
---
 tests/test_tipc/prepare.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_tipc/prepare.sh b/tests/test_tipc/prepare.sh
index e57feda0..7d4dd8b1 100755
--- a/tests/test_tipc/prepare.sh
+++ b/tests/test_tipc/prepare.sh
@@ -35,6 +35,8 @@ if [[ ${MODE} = "benchmark_train" ]];then
     pip install setuptools_scm #-i https://pypi.tuna.tsinghua.edu.cn/simple 
     pip install . #-i https://pypi.tuna.tsinghua.edu.cn/simple 
     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
     cd -
     if [[ ${model_name} == "conformer" ]]; then