在pytorch实现机器学习代码时,self.children()的作用

self.children()是一个方法,它返回模型(通常是一个nn.Module的实例)中所有子模块的迭代器。这些子模块可以是层(如nn.Conv2dnn.ReLU等),也可以是包含其他子模块的容器(如nn.Sequential、自定义的nn.Module子类)。

当你定义一个复杂的神经网络模型时,self.children()可以帮助你遍历模型中的所有层,这在某些情况下很有用:

1)在模型的前向传播中,你可能想要以某种方式迭代处理所有的子层。

2)在模型的初始化阶段,你可能需要为特定的子层设置特定的参数初始化策略。

3)在模型的训练过程中,你可能想要对不同的子层应用不同的正则化或优化策略。

例子:

import torch.nn as nn  
  
class SimpleCNN(nn.Module):  
    def __init__(self):  
        super(SimpleCNN, self).__init__()  
        self.conv1 = nn.Conv2d(3, 16, kernel_size=3, stride=1, padding=1)  
        self.relu1 = nn.ReLU()  
        self.conv2 = nn.Conv2d(16, 32, kernel_size=3, stride=1, padding=1)  
        self.relu2 = nn.ReLU()  
        self.pool = nn.MaxPool2d(kernel_size=2, stride=2)  
        self.fc1 = nn.Linear(32 * 7 * 7, 128)  
        self.relu3 = nn.ReLU()  
        self.fc2 = nn.Linear(128, 10)  
  
    def forward(self, x):  
        x = self.conv1(x)  
        x = self.relu1(x)  
        x = self.conv2(x)  
        x = self.relu2(x)  
        x = self.pool(x)  
        x = x.view(-1, 32 * 7 * 7)  
        x = self.fc1(x)  
        x = self.relu3(x)  
        x = self.fc2(x)  
        return x  
  
# 实例化模型  
model = SimpleCNN()  
  
# 遍历所有子层  
for child in model.children():  
    print(type(child))  
    # 你可以在这里对child执行任何操作,例如检查其参数、打印输出尺寸等

输出结果如下

如果我们把最后一句输出代码改成print(child),则会输出所有层的结构

如果把输出修改为print(child.name),因为我们没有给每层命名,所以会报错:

  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝海渔夫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值