update

Oneflow-Inc · Nov 10, 2023 · 2f6956a · 2f6956a
1 parent 4d29235
commit 2f6956a
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -58,20 +58,21 @@ jobs:
             --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
             -v /share_nfs/dataset/ImageNet:/data/dataset/ImageNet \
             -v /share_nfs/model_zoo/flowvision_cache:/flowvision_cache \
+            -e FLOWVISION_CACHE=/flowvision_cache \
             -v $(pwd)/${{ env.TEST_DIR }}:/workspace/${{ env.TEST_DIR }} \
             -w /workspace/${{ env.TEST_DIR }} \
             --name ${{ env.TEST_CONTAINER_NAME }} \
             ${{ env.TEST_IMG_TAG }} \
             sleep 5400
-      - name: ls
-        run: |
-          docker exec ${{ env.TEST_CONTAINER_NAME }} ls /flowvision_cache
       - name: run inference
         run: |
-          docker exec ${{ env.TEST_CONTAINER_NAME }} FLOWVISION_CACHE=/flowvision_cache bash ./infer.sh
+          docker exec ${{ env.TEST_CONTAINER_NAME }} ls /flowvision_cache
+          docker exec ${{ env.TEST_CONTAINER_NAME }} bash ./infer.sh
+          #docker exec ${{ env.TEST_CONTAINER_NAME }} FLOWVISION_CACHE=/flowvision_cache bash ./infer.sh
       - 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 }} FLOWVISION_CACHE=/flowvision_cache bash ./train.sh
+          docker exec ${{ env.TEST_CONTAINER_NAME }} bash ./train.sh
+          #docker exec ${{ env.TEST_CONTAINER_NAME }} FLOWVISION_CACHE=/flowvision_cache bash ./train.sh