Unverified Commit 232214b2 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Bugfix] Replace `PoolingParams.normalize` with `use_activation` (#32243)


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent eb28e806
...@@ -801,7 +801,7 @@ def get_pooling_config( ...@@ -801,7 +801,7 @@ def get_pooling_config(
logger.info("Found pooling configuration.") logger.info("Found pooling configuration.")
config: dict[str, Any] = {"normalize": normalize} config: dict[str, Any] = {"use_activation": normalize}
for key, val in pooling_dict.items(): for key, val in pooling_dict.items():
if val is True: if val is True:
pooling_type = parse_pooling_type(key) pooling_type = parse_pooling_type(key)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment