diff options
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 97cc8736..19080926 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -260,7 +260,12 @@ def main(args=None): choices=[True, False], help="Control if Softmax should be transformed into a set of npu operations (default: %(default)s)", ) - + parser.add_argument( + "--weight-estimation-scaling", + type=float, + default=1.0, + help=("Performs an additional scaling of weight compression scale estimate (default: %(default)s)"), + ) args = parser.parse_args(args=args) # Read configuration file @@ -291,6 +296,7 @@ def main(args=None): global_memory_clock_scale=args.global_memory_clock_scale, max_blockdep=args.max_block_dependency, softmax_support=args.softmax_support, + weight_estimation_scaling=args.weight_estimation_scaling, ) compiler_options = compiler_driver.CompilerOptions( |