We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
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。
非常感谢!
The text was updated successfully, but these errors were encountered:
GitHub-HongweiZhang
No branches or pull requests
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。
非常感谢!
The text was updated successfully, but these errors were encountered: