Hello, thank you for your great work!
I have an issue when I want to send nn.parameter to the optimizer, the error is as follow:
TypeError: params argument given to the optimizer should be an iterable of Tensors or dicts, but got torch.cuda.FloatTensor
Thank you!