About the code in DGL

Open MrZealand opened this issue 6 years ago • 3 comments

The code in https://github.com/dmlc/dgl/blob/master/examples/pytorch/han/model.py Line 19,Line21. beta = torch.softmax(w, dim=1) return (beta * z).sum(1)

The dim should be 0 ?

Apr 13 '20 08:04 MrZealand

The features are on dim=1 . So it is correct.

May 06 '20 20:05 mongoose54

I found code in dgl obtains inferior performance. Do you have any idea? Please ref this issue! Thanks if you have time

Dec 02 '20 06:12 lipingcoding

@lipingcoding what doo you mean by performance? speed or final accuracy?

Dec 11 '20 18:12 yzh119