From c31a5eaf79aa48d1f1a7fbdb5177fe26ae190a88 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Fri, 23 May 2025 10:10:07 -0400 Subject: [PATCH] removed warnings from training --- .riahub/workflows/workflow.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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