pytorch 查看优化optimizer其中参数的学习率

from torchvision.models import resnet18
import torch

model = resnet18()
param = torch.zeros([2,2])
optimizer = torch.optim.Adam([param],lr=1e-4)
for param_group in optimizer.param_groups:# 
    print(param_group) 

output:{'params': [tensor([[0., 0.],
        [0., 0.]])], 'lr': 0.0001, 'betas': (0.9, 0.999), 'eps': 1e-08, 'weight_decay': 0, 'amsgrad': False}

param_group是个字典，其中有’params’存储参数，'lr’存储学习率