Upload train.py in train_workflow.yml

gpu-mode · Nov 14, 2024 · b7f441c · b7f441c
1 parent 1c4f2ed
commit b7f441c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
@@ -46,11 +46,13 @@ jobs:
       run: |
         python train.py > training.log 2>&1
 
-    - name: Upload logs
+    - name: Upload artifacts
       uses: actions/upload-artifact@v3
       if: always()
       with:
-        name: training-logs
-        path: training.log
+        name: training-artifacts
+        path: |
+          training.log
+          train.py
       env:
         CUDA_VISIBLE_DEVICES: 0 # Make sure only one GPU is used for testing