Yes I am calling the parameters of that particular tensor for specifying the learning rate something like this:
optimizer = optim.Adam([{'params':model.custom_param.parameters(), 'lr':1e-3}, ... and so on])
How can I do that?
Yes I am calling the parameters of that particular tensor for specifying the learning rate something like this:
optimizer = optim.Adam([{'params':model.custom_param.parameters(), 'lr':1e-3}, ... and so on])
How can I do that?