This commit is contained in:
parent
91bab4acbd
commit
42e7429f02
|
@ -120,8 +120,7 @@ jobs:
|
||||||
|
|
||||||
- name: 6. Convert to ORT file
|
- name: 6. Convert to ORT file
|
||||||
run: |
|
run: |
|
||||||
python -m scripts/ort/convert_to_ort.py
|
PYTHONPATH=. python scripts/ort/convert_to_ort.py
|
||||||
|
|
||||||
|
|
||||||
- name: Upload ORT file
|
- name: Upload ORT file
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
|
|
|
@ -2,7 +2,7 @@ import os
|
||||||
import torch
|
import torch
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from sklearn.metrics import classification_report
|
from sklearn.metrics import classification_report
|
||||||
import matplotlib
|
os.environ["NNPACK"] = "0"
|
||||||
from matplotlib import pyplot as plt
|
from matplotlib import pyplot as plt
|
||||||
|
|
||||||
from scripts.training.mobilenetv3 import mobilenetv3, RFClassifier
|
from scripts.training.mobilenetv3 import mobilenetv3, RFClassifier
|
||||||
|
|
Loading…
Reference in New Issue
Block a user