Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

wbwtab下的bn_fuse 设置问题 #109

Open
cqray1990 opened this issue Jan 5, 2023 · 0 comments
Open

wbwtab下的bn_fuse 设置问题 #109

cqray1990 opened this issue Jan 5, 2023 · 0 comments

Comments

@cqray1990
Copy link

wbwtab下的bn_fuse 设置如何参数,都是32bit的bn融合,
parser.add_argument("--W", type=int, default=32, help="Wb:2, Wt:3, Wfp:32")
parser.add_argument("--A", type=int, default=32, help="Ab:2, Afp:32")

为啥代码还是进入:

******************* 针对特征(A)二值的bn融合 *******************

if bn_counter >= 1 and bn_counter <= bin_bn_fuse_num:
    mask_positive = gamma.data.gt(0)
    mask_negetive = gamma.data.lt(0)

    w_fused[mask_positive] = w[mask_positive]
    b_fused[mask_positive] = (
        b[mask_positive]
        - mean[mask_positive]
        + beta[mask_positive] * (std[mask_positive] / gamma[mask_positive])
    )

融合二值的,这种设置应该是普通融合啊?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant