PyTorch参数自动命名规则

1 · HikariLi · Jan. 23, 2024, 3:43 a.m.
Summary
PyTorch参数自动命名规则当我们使用12for name, params in model.named_parameters(): print(f"Parameter Name: {name}, Parameter Shape: {params.shape}")时可以看到模型的参数以及参数的名字,PyTorch内部实际上有一套命名的规则。我们创建一个简单的CNN模型123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475class Block(nn.Module): def __init__(self, channels): super().__init__() self.conv1 = nn.Conv2d( in_channels=channels, out_channel...