Pytorch——查找、替换module相关操作

nn.Module类可用操作

1. model.named_parameters()

python 复制代码
# 遍历模型的所有参数并打印它们的名称和形状
for name, param in model.named_parameters():
    print(f"Parameter Name: {name}, Parameter Shape: {param.shape}")

输出示例:

python 复制代码
Parameter Name: conv1.weight, Parameter Shape: torch.Size([64, 3, 3, 3])
Parameter Name: conv1.bias, Parameter Shape: torch.Size([64])
Parameter Name: conv2.weight, Parameter Shape: torch.Size([64, 64, 3, 3])
Parameter Name: conv2.bias, Parameter Shape: torch.Size([64])
Parameter Name: fc.weight, Parameter Shape: torch.Size([10, 64])
Parameter Name: fc.bias, Parameter Shape: torch.Size([10])

2. model.named_modules()

python 复制代码
# 遍历模型的所有模块并打印它们的名称和类型
for name, module in model.named_modules():
    print(f"Module Name: {name}, Module Type: {module.__class__.__name__}")

输出示例:

python 复制代码
Module Name: , Module Type: MyModel
Module Name: conv1, Module Type: Conv2d
Module Name: relu, Module Type: ReLU
Module Name: conv2, Module Type: Conv2d
Module Name: fc, Module Type: Linear

3. model.get_submodule()

python 复制代码
# 遍历模型的所有模块并打印它们的名称和类型
for name, module in model.named_modules():
	# 通过name获取子模块
	sub_module = model.get_submodule(name)
    print(f"Module Name: {name}, Module Type: {sub_module.__class__.__name__}")

输出示例:

python 复制代码
Module Name: , Module Type: MyModel
Module Name: conv1, Module Type: Conv2d
Module Name: relu, Module Type: ReLU
Module Name: conv2, Module Type: Conv2d
Module Name: fc, Module Type: Linear
相关推荐
凌览28 分钟前
有了 25k Star 的MediaCrawler爬虫库加持,三分钟搞定某红书、某音等平台爬取!
前端·后端·python
G.E.N.28 分钟前
开源!RAG竞技场(2):标准RAG算法
大数据·人工智能·深度学习·神经网络·算法·llm·rag
西西弗Sisyphus38 分钟前
如果让计算机理解人类语言- Word2Vec(Word to Vector,2013)
人工智能·word·word2vec
这里有鱼汤39 分钟前
给你的DeepSeek装上实时行情,让他帮你炒股
后端·python·mcp
前端双越老师1 小时前
30 行代码 langChain.js 开发你的第一个 Agent
人工智能·node.js·agent
Tomorrow'sThinker1 小时前
[特殊字符] Excel 读取收件人 + Outlook 批量发送带附件邮件 —— Python 自动化实战
python·excel·outlook
东坡肘子1 小时前
高温与奇怪的天象 | 肘子的 Swift 周报 #092
人工智能·swiftui·swift
JosieBook1 小时前
【Java编程动手学】Java常用工具类
java·python·mysql
Green1Leaves1 小时前
pytorch学习-11卷积神经网络(高级篇)
pytorch·学习·cnn
KaneLogger2 小时前
视频转文字,别再反复拖进度条了
前端·javascript·人工智能