From 09ecc5c8acb758e8def33155feb746a34dd7b560 Mon Sep 17 00:00:00 2001 From: Annie Tallund Date: Wed, 14 Dec 2022 15:55:19 +0100 Subject: MLIA-590 Support path to custom target profiles - Start using TOML format for target profile - Add support for loading custom target profile files Change-Id: I6be019d4341e93115440ccdbdb6dafdc1c85b966 --- src/mlia/target/ethos_u/advisor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/mlia/target/ethos_u') diff --git a/src/mlia/target/ethos_u/advisor.py b/src/mlia/target/ethos_u/advisor.py index ce4e0fc..937e91c 100644 --- a/src/mlia/target/ethos_u/advisor.py +++ b/src/mlia/target/ethos_u/advisor.py @@ -123,7 +123,7 @@ class EthosUInferenceAdvisor(DefaultInferenceAdvisor): def configure_and_get_ethosu_advisor( context: ExecutionContext, - target_profile: str, + target_profile: str | Path, model: str | Path, **extra_args: Any, ) -> InferenceAdvisor: @@ -155,7 +155,7 @@ _DEFAULT_OPTIMIZATION_TARGETS = [ def _get_config_parameters( model: str | Path, - target_profile: str, + target_profile: str | Path, **extra_args: Any, ) -> dict[str, Any]: """Get configuration parameters for the advisor.""" -- cgit v1.2.1