feat: 添加一键运行脚本 run_all.py
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
49
vsp/qwen3.5-9b/run_all.py
Normal file
49
vsp/qwen3.5-9b/run_all.py
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
"""一键运行所有测试"""
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
|
||||||
|
|
||||||
|
SCRIPTS = [
|
||||||
|
("环境检查", "vsp/qwen3.5-9b/setup_env.py"),
|
||||||
|
("模型下载", "vsp/qwen3.5-9b/download_model.py"),
|
||||||
|
("基础推理测试", "vsp/qwen3.5-9b/test_basic_inference.py"),
|
||||||
|
("性能基准测试", "vsp/qwen3.5-9b/benchmark_speed.py"),
|
||||||
|
("精度评估", "vsp/qwen3.5-9b/test_accuracy.py"),
|
||||||
|
("并发压测", "vsp/qwen3.5-9b/test_concurrency.py"),
|
||||||
|
("GPU需求分析", "vsp/qwen3.5-9b/gpu_requirements.py"),
|
||||||
|
("生成报告", "vsp/qwen3.5-9b/generate_report.py"),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
os.chdir(os.path.dirname(os.path.abspath(__file__)) + "/../..")
|
||||||
|
print("=" * 60)
|
||||||
|
print("Qwen3.5-9B 全量测试")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
for name, script in SCRIPTS:
|
||||||
|
print(f"\n{'='*60}")
|
||||||
|
print(f"[{name}] 运行 {script}")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
t0 = time.time()
|
||||||
|
result = subprocess.run([sys.executable, script], capture_output=False)
|
||||||
|
elapsed = time.time() - t0
|
||||||
|
|
||||||
|
if result.returncode != 0:
|
||||||
|
print(f"\n[ERROR] {name} 失败 (退出码: {result.returncode})")
|
||||||
|
choice = input("继续运行后续测试?(y/n): ").strip().lower()
|
||||||
|
if choice != "y":
|
||||||
|
sys.exit(1)
|
||||||
|
else:
|
||||||
|
print(f"\n[OK] {name} 完成 ({elapsed:.1f}s)")
|
||||||
|
|
||||||
|
print(f"\n{'='*60}")
|
||||||
|
print("所有测试完成!查看报告: vsp/qwen3.5-9b/results/REPORT.md")
|
||||||
|
print("=" * 60)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
Reference in New Issue
Block a user