gpu1

Oneflow-Inc · Nov 10, 2023 · edfae1d · edfae1d
1 parent 665a37e
commit edfae1d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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