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

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

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 8930f5b8..198d56f4 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -71,6 +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 }} bash ./train.sh