Open YingHH1 opened 2 years ago
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。
非常感谢!
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。
非常感谢!