diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml index c4714ef..bc0f304 100644 --- 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