DIFF.BLOG
New
Following
Discover
Jobs
More
Top Writers
Suggest a blog
Upvotes plugin
Report bug
Contact
About
Sign up  
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...
Read full post on jesseprince.github.io →
Submit
AUTHOR
RECENT POSTS FROM THE AUTHOR