From fa123c1adc827552a14cf5982779f6495f833eec Mon Sep 17 00:00:00 2001
From: ShawnXuan <xiexuanx2@gmail.com>
Date: Fri, 10 Nov 2023 16:00:57 +0800
Subject: [PATCH] gpu1

---
 .github/workflows/test.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 198d56f4..572979d3 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -71,7 +71,7 @@ jobs:
       - name: run training
         run: |
           docker exec ${{ env.TEST_CONTAINER_NAME }} sed -i 's/main.py/main.py --epochs 1/g' train.sh 
-          docker exec ${{ env.TEST_CONTAINER_NAME }} sed -i 's/GPU_NUMS=8/GPU_NUMS=1/g' train.sh 
+          docker exec ${{ env.TEST_CONTAINER_NAME }} sed -i 's/GPU_NUMS=[0-9]*/GPU_NUMS=2/g' train.sh 
           docker exec ${{ env.TEST_CONTAINER_NAME }} bash ./train.sh