diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
index 034f652..3814d15 100644
--- 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