search

beta-team / beta-recsys

Beta-RecSys: Build, Evaluate and Tune Automated Recommender Systems
https://beta-recsys.readthedocs.io/en/latest/
MIT License
163 stars 33 forks source link

Improve code #413

Closed TedSIWEILIU closed 3 years ago

TedSIWEILIU commented 3 years ago
  1. remove double initialise the train engine for lightgcn
  2. replace "nn.init.normal" with "nn.init.normal_"