remove warnings

This commit is contained in:
Liyu Xiao 2025-06-18 13:52:10 -04:00
parent 91bab4acbd
commit 42e7429f02
2 changed files with 2 additions and 3 deletions

View File

@ -120,8 +120,7 @@ jobs:
- name: 6. Convert to ORT file
run: |
python -m scripts/ort/convert_to_ort.py
PYTHONPATH=. python scripts/ort/convert_to_ort.py
- name: Upload ORT file
uses: actions/upload-artifact@v3

View File

@ -2,7 +2,7 @@ import os
import torch
import numpy as np
from sklearn.metrics import classification_report
import matplotlib
os.environ["NNPACK"] = "0"
from matplotlib import pyplot as plt
from scripts.training.mobilenetv3 import mobilenetv3, RFClassifier