optim:
torch.optim.SGD([
{'params': model.base.parameters()},
{'params': model.classifier.parameters(), 'lr': 1e-3}
], lr=1e-2, momentum=0.9)
Lambda LR:
torch.optim.lr_scheduler.LambdaLR(optimizer,lr_lambda,last_epoch=-1)
torch.optim.SGD([
{'params': model.base.parameters()},
{'params': model.classifier.parameters(), 'lr': 1e-3}
], lr=1e-2, momentum=0.9)
torch.optim.lr_scheduler.LambdaLR(optimizer,lr_lambda,last_epoch=-1)