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

zero_module #152

Open
MsQueeniee opened this issue Jul 3, 2024 · 0 comments
Open

zero_module #152

MsQueeniee opened this issue Jul 3, 2024 · 0 comments

Comments

@MsQueeniee
Copy link

为什么要在每一个block中使用zero_module将网络参数清零

Why do we need to use zero_module to zero out the network parameters in each block?

def zero_module(module):
"""
Zero out the parameters of a module and return it.
"""
for p in module.parameters():
p.detach().zero_()
return module

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