From 146bc8cb82f0d2eb799561a50df0ed92793bca2e Mon Sep 17 00:00:00 2001
From: Ben Horowitz <bhorowit@gmail.com>
Date: Thu, 14 Nov 2024 20:37:32 -0800
Subject: [PATCH] Upload train.py in train_workflow.yml (#22)

---
 .github/workflows/train_workflow.yml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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