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

模型参数量计算 #44

Open
zxx20231119 opened this issue Nov 19, 2023 · 2 comments
Open

模型参数量计算 #44

zxx20231119 opened this issue Nov 19, 2023 · 2 comments

Comments

@zxx20231119
Copy link

max_seq_len = 512
dim = 512
n_layers = 8
n_heads = 8

改变值后,如何计算最终的参数量,有公式吗?

@zerozhoujie
Copy link

在model.py文件的268行可以打印一下模型都有哪些层,以及每一层的参数量

@zxx20231119
Copy link
Author

好的,谢谢

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

2 participants