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

RWKV7 gate更新方式代码和论文不一致 #280

Open
jihaoh98 opened this issue Dec 27, 2024 · 4 comments
Open

RWKV7 gate更新方式代码和论文不一致 #280

jihaoh98 opened this issue Dec 27, 2024 · 4 comments

Comments

@jihaoh98
Copy link

你好,看起来g的更新方式代码和论文存在差异,代码中是g = torch.sigmoid(xg @ self.g1) @ self.g2,但是在结构图中是g = torch.sigmoid(xg @ self.g1 @ self.g2)写法,请问哪里存在错误?

@BlinkDL
Copy link
Owner

BlinkDL commented Dec 27, 2024

谢谢,请以代码为准,我告诉结构图作者

@BlinkDL
Copy link
Owner

BlinkDL commented Dec 27, 2024

@jihaoh98 请再检查还有没有错,多谢多谢

@BlinkDL
Copy link
Owner

BlinkDL commented Dec 31, 2024

https://rwkv.com 已更新

@jihaoh98
Copy link
Author

thanks!

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