if args.bottleneck:
     optimizer = torch.optim.SGD([
         {'params': model.feature_layers.parameters()},
         {'params': model.bottle.parameters(), 'lr': args.lr[1]},
         {'params': model.cls_fc.parameters(), 'lr': args.lr[2]},
     ], lr=args.lr[0], momentum=args.momentum, weight_decay=args.decay)

最后传入的学习率以数组的方式呈现

parser.add_argument('--lr', type=list, help='Learning rate', default=[0.001, 0.01, 0.01])