Hello, that’s the only error message I got.
so basically instead of using model.parameters(), I send it my own parameter list, parameter = [] into the optimizer.
I tried adding self.U.parameters() into the list and optimize it, the error was “optimizer can only optimize Tensors, but one of the params is Module.parameters”…