xh_hit的博客

一般网络使用pytorch训练，权重保存在pth中。读取pth使用state_dict = torch.load('path/*.pth', map_location='cpu')得到的state_dict是一个字典，key为层的name，value为weight。假设block1为由conv2d，BN组成的，那么卷积对应的key为block1.0.weight，block1.0.bias，BN层参数较多，包括running_mean ，running_var，gamma，beta，eps，...