search

georgeguo-cn / LGMRec

[AAAI 2024] Official Pytorch Implementation for "LGMRec: Local and Global Graph Learning for Multimodal Recommendation".
22 stars 7 forks source link

Question about Model #2

Open Tianyi-Billy-Ma opened 2 months ago

Tianyi-Billy-Ma commented 2 months ago

In the file models/lgmrec.py line 118, why assign the gumble_softmax value of ut_hyper to uv_hyper?

The source code is uv_hyper = F.gumbel_softmax(ut_hyper, self.tau, dim=1, hard=False).

Isn't this should be ut_hyper = F.gumbel_softmax(ut_hyper, self.tau, dim=1, hard=False)?

georgeguo-cn commented 2 months ago

Thank you for your thorough inspection. You are right and I have made the following modifications:

ut_hyper = F.gumbel_softmax(ut_hyper, self.tau, dim=1, hard=False).