New Model

[Masoumeh-Alinia]

Hello, first of all thanks for your great work with the library! I had a question. I want to use the library's defined model from LightGCN and add another series of layers and then train the model. It is possible? How can I do this because the input for the next layer must be tensor.

[massquantity]

Hi, you can inherit from the LightGCN class. Internally it uses a pytorch module self.torch_model to perform the forward propagation. So you can add layers after this module.