feat: Add ResNet50 benchmark script with CI/CD artifact collection

.gitlab-ci.yml

+# 定义流水线的阶段
+stages:
+  - test
+  - deploy
+
+# === 作业 1: 测试阶段 (运行压测) ===
+benchmark_job:
+  stage: test
+  # 关键：指定使用我们注册的 Runner
+  tags:
+    - test
+  script:
+    - echo "Starting ResNet50 benchmark..."
+    # 直接在 Runner 容器的 shell 中执行 Python 脚本
+    - python benchmark_resnet50.py
+    - echo "Benchmark finished."
+  # 定义产物：将脚本生成的 results.json 文件保存下来
+  artifacts:
+    paths:
+      - results.json
+    # 产物保存 1 周，过期自动删除
+    expire_in: 1 week
+
+# === 作业 2: 部署阶段 (展示结果) ===
+# 这个作业的目的是展示如何使用上一个作业的产物
+deploy_job:
+  stage: deploy
+  tags:
+    - test
+  # 只在主分支上运行，模拟一个正式的部署或报告流程
+  rules:
+    - if: $CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH
+  # GitLab 会自动将上一个作业 (benchmark_job) 的产物下载到此作业的工作目录中
+  script:
+    - echo "Displaying benchmark results from the artifact..."
+    - cat results.json
+    - echo "Results have been archived."
+  environment:
+    name: performance-report
+    # 这个链接会指向 CI/CD 的作业列表，方便追溯
+    url: https://gitlab.com/your-username/resnet50-benchmark/-/jobs

benchmark_resnet50.py

+import torch
+import torchvision.models as models
+import time
+import json
+
+def run_resnet50_benchmark():
+    """运行 ResNet50 推理基准测试，batch_size=64"""
+    print("--- PyTorch ResNet50 Benchmark (batch_size=64) ---")
+
+    # 1. 检查设备 (CPU or GPU)
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    print(f"Running on device: {device}")
+
+    # 2. 加载预训练的 ResNet50 模型
+    print("Loading ResNet50 model with pretrained weights...")
+    model = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V2).to(device)
+    # 设置为评估模式，关闭 Dropout 等训练层
+    model.eval()
+
+    # 3. 准备输入数据，batch_size=64
+    batch_size = 64
+    # ImageNet 标准输入尺寸: (batch_size, channels, height, width)
+    input_tensor = torch.randn(batch_size, 3, 224, 224).to(device)
+    print(f"Input tensor shape: {input_tensor.shape}")
+
+    # 4. 预热 (非常重要！避免首次运行的初始化开销影响结果)
+    print("Warming up...")
+    with torch.no_grad():  # 关闭梯度计算，节省内存和计算
+        for _ in range(10):
+            _ = model(input_tensor)
+
+    # 5. 正式进行基准测试
+    print("Running benchmark...")
+    num_runs = 100
+    start_time = time.time()
+
+    with torch.no_grad():
+        for _ in range(num_runs):
+            _ = model(input_tensor)
+
+    end_time = time.time()
+
+    # 6. 计算性能指标
+    total_time = end_time - start_time
+    total_images = num_runs * batch_size
+    
+    # 吞吐量: 每秒处理的图片数量
+    throughput = total_images / total_time
+    
+    # 平均延迟: 处理单张图片的平均时间 (毫秒)
+    avg_latency_ms = (total_time / total_images) * 1000
+
+    # 7. 整理结果
+    results = {
+        "model": "ResNet50",
+        "device": str(device),
+        "batch_size": batch_size,
+        "num_runs": num_runs,
+        "total_time_s": round(total_time, 4),
+        "throughput_imgs_per_sec": round(throughput, 2),
+        "avg_latency_ms_per_img": round(avg_latency_ms, 4)
+    }
+
+    # 8. 打印并保存结果到 JSON 文件
+    print("\n--- Benchmark Results ---")
+    print(json.dumps(results, indent=4))
+    
+    with open("results.json", "w") as f:
+        json.dump(results, f, indent=4)
+        
+    print(f"\nBenchmark finished. Results saved to results.json")
+
+
+if __name__ == "__main__":
+    run_resnet50_benchmark()