diff --git a/scripts/application_packager/convert_to_onnx.py b/scripts/application_packager/convert_to_onnx.py index c99a265..f022c39 100644 --- a/scripts/application_packager/convert_to_onnx.py +++ b/scripts/application_packager/convert_to_onnx.py @@ -2,7 +2,7 @@ import os import numpy as np import torch -from scripts.training.mobilenetv3 import RFClassifier, mobilenetv3 +from scripts.model_builder.mobilenetv3 import RFClassifier, mobilenetv3 from helpers.app_settings import get_app_settings