removed warnings from training

2025-05-23 10:10:07 -04:00 · 2025-05-23 10:10:07 -04:00 · c31a5eaf79
commit c31a5eaf79
parent 473dc5bb40
1 changed files with 2 additions and 1 deletions
--- a/.riahub/workflows/workflow.yaml
+++ b/.riahub/workflows/workflow.yaml
@ -47,8 +47,9 @@ jobs:
      - name: 2. Train Model
        env:
            NO_NNPACK: 1
+            PYTORCH_NO_NNPACK: 1
        run: |
-          PYTHONPATH=. python data/training/train.py
+          PYTHONPATH=. python data/training/train.py 2>/dev/null
          echo "training model"

      - name: 3. Build inference app