Thank you for your response.
However, while the assignment works, when passing the parameters to initialize the optimizer, the following error is returned.
ValueError: can’t optimize a non-leaf Tensor
I am wondering what is the reason behind this, and how to solve the error.