diff options
Diffstat (limited to 'src/mlia/target/config.py')
-rw-r--r-- | src/mlia/target/config.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/src/mlia/target/config.py b/src/mlia/target/config.py index 8492086..8a5b360 100644 --- a/src/mlia/target/config.py +++ b/src/mlia/target/config.py @@ -71,7 +71,16 @@ def is_builtin_target_profile(profile_name: str | Path) -> bool: return profile_name in BUILTIN_SUPPORTED_PROFILE_NAMES -BUILTIN_SUPPORTED_OPTIMIZATION_NAMES = ["optimization"] +BUILTIN_SUPPORTED_OPTIMIZATION_NAMES = [ + "optimization", + "optimization-custom-augmentation", + "optimization-fully-connected-clustering", + "optimization-fully-connected-pruning", + "optimization-fully-connected-unstructured-pruning", + "optimization-conv2d-clustering", + "optimization-conv2d-pruning", + "optimization-conv2d-unstructured-pruning", +] def is_builtin_optimization_profile(optimization_name: str | Path) -> bool: |