Hi alban,
self is not a module in my case, I have tried using self.U.parameters() and the error is now
“optimizer can only optimize Tensors, but one of the params is Module.parameters”
Do you have any idea why is this so?
I was trying to write a customisable program so I didn’t inherit nn.Module, thank you!