training doesn't seem to work

triton-lang · Oct 17, 2023 · f7bf652 · f7bf652
1 parent fa15c31
commit f7bf652
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/torchbench.yml b/.github/workflows/torchbench.yml
@@ -49,7 +49,7 @@ jobs:
       - name: Run TorchBench
         run: |
           cd torchbench
-          python3 run.py BERT_pytorch -d cuda -t train --torchdynamo inductor
+          python3 run.py BERT_pytorch -d cuda -t eval --torchdynamo inductor
 
       - name: Install the latest Triton wheel
         run: |
@@ -59,7 +59,7 @@ jobs:
       - name: Run TorchBench Again
         run: |
           cd torchbench
-          python3 run.py BERT_pytorch -d cuda -t train --torchdynamo inductor
+          python3 run.py BERT_pytorch -d cuda -t eval --torchdynamo inductor
   
       - name: Cleanup and reset GPU clocks
         if: always()