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
相关推荐
zm-v-159304339869 分钟前
ArcGIS 水文分析升级:基于深度学习的流域洪水演进过程模拟
人工智能·深度学习·arcgis
拓端研究室1 小时前
视频讲解|核密度估计朴素贝叶斯:业务数据分类—从理论到实践
人工智能·分类·数据挖掘
灵智工坊LingzhiAI1 小时前
人体坐姿检测系统项目教程(YOLO11+PyTorch+可视化)
人工智能·pytorch·python
昨日之日20061 小时前
Video Background Remover V3版 - AI视频一键抠像/视频换背景 支持50系显卡 一键整合包下载
人工智能·音视频
SHIPKING3932 小时前
【机器学习&深度学习】什么是下游任务模型?
人工智能·深度学习·机器学习
子燕若水6 小时前
Unreal Engine 5中的AI知识
人工智能
极限实验室7 小时前
Coco AI 实战(一):Coco Server Linux 平台部署
人工智能
杨过过儿7 小时前
【学习笔记】4.1 什么是 LLM
人工智能
巴伦是只猫7 小时前
【机器学习笔记Ⅰ】13 正则化代价函数
人工智能·笔记·机器学习
大千AI助手8 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配