I think the problem I am having is not inheriting nn.Module in my class. Therefore, the method .parameters() is not available to me.
Is there any other way that I could change nn.Embedding into something that the optimizer can understand? So that I can add it into the parameter list manually?
Thank you