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

Possibility of adding DICE to replace prelu? Also, a small bug for the GPU implementation for newer PyTorch versions #35

Open
YingHH1 opened this issue Feb 14, 2022 · 0 comments
Assignees
Labels
bug Something isn't working

Comments

@YingHH1
Copy link

YingHH1 commented Feb 14, 2022

Hi,

谢谢大佬的code。想问一下在DIN/DIEN里是否会在未来加入DICE激活函数来取代torch自身提供的prelu?

目前发现的小问题,在gpu运行下(torch 1.9.0):
在DIEN.py里因为pytorch版本更新带来的一些new behaviour,需要加入
keys_length=keys_length.to(torch.device("cpu"))
作为torch里的改动要求(需要非cuda的cpu long type)。之后在interest.py里加入
keys_length=keys_length.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
来改回为GPU。

非常感谢!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants