AI Pool
Log in
Discussions>How to have different learning rates for the parts of a network in Pytorch?>

How to have different learning rates for the parts of a network in Pytorch?

I've got a segmentation network, which contains an encoder and multiple decoders. I want to have different learning rates for each part. How can I make it happen?

Computer VisionAI Development/Engineering
learning-ratepytorch
3 votesJW326.00
1 Answers
JO295.00
3

First, you need to define those parts in different classes to have different names. After that, you just filter and collect those parts by their layer-names.

encoder = []
decoder = []
for name, param in net.named_parameters():
  if 'encoder' in name:
    encoder.append(param)
  elif 'decoder' in name:
    decoder.append(param)

Then you pass those params to the optimizer separately, where you can treat them separately too.

optimizer = torch.optim.SGD([{'params':encoder}, {'params':decoder}], lr=DEFOULT_LRATE, nesterov=True)
...
optimizer.param_groups[0]['lr'] = L_RATE1
optimizer.param_groups[1]['lr'] = L_RATE2
Reply
Couldn't find what you were looking for?and we will find an expert to answer.
How helpful was this page?