2024年7月1日
使用不同函数打印torch.nn模型——print(model),named_children(),named_modules():
摘要： 创建模型 创建一个具有三级嵌套的模型，结构如图： import torch import torch.nn as nn # 定义子子模块 class SubSubModule(nn.Module): def __init__(self): super(SubSubModule, self).__in 阅读全文
posted @ 2024-07-01 16:34 seekwhale13 阅读(43) 评论(0) 推荐(0) 编辑