Model.modules和Model.children

Model.modules和Model.children
首先我们先定义一个网络结构：

class Linear(nn.Module):def __init__(self):super().__init__()self.encoder = nn.Sequential(nn.Conv2d(),nn.BatchNorm2d(),nn.ReLU())self.decoder = nn.Sequential(nn.Linear(),nn.Interpolate())def forward(self,x)output = self.encoder(x)output = self.decoder(output)return output

然后我们实例化类：

model = Linear()

对于Model.modules，我们遍历整个模型：

会得到：
首先从linear层开始遍历：
nn.Sequential(
nn.Conv2d(),
nn.BatchNorm2d(),
nn.ReLU）

nn.Sequential(
nn.Linear(),
nn.Interpolate())
接着从self.encoder，self.decoder层开始向下遍历：
nn.Sequential(
nn.Conv2d(),
nn.BatchNorm2d(),
nn.ReLU）

nn.Conv2d(),
nn.BatchNorm2d(),
nn.ReLU

nn.Sequential(
nn.Linear(),
nn.Interpolate())

nn.Linear(),
nn.Interpolate()

而Model.children只会遍历self.encoder和self.encoder:
nn.Sequential(
nn.Conv2d(),
nn.BatchNorm2d(),
nn.ReLU())
nn.Sequential(
nn.Linear(),
nn.Interpolate())

Model.modules和Model.children相关推荐

Pytorch中的model.modules()和model.children()的区别
Pytorch中的model.modules()和model.children()的区别背景:最近在做网络模型中可视化的过程中,需要将网络结构中的某一层的特征进行输出.所以就遇到了这个问题,小小记录 ...
model.modules()和model.children()以及model._modules.items()
两者包含的其实是你所定义的网络中继承自nn.Module的模块,比如nn.Relu,nn.Conv2d等,并不是你整个网络的forward()结构,而且你自定义的函数模块也是没有的,实验如下: 另外, ...
pytroch：model.train()、model.eval()的使用
前言:最近在把两个模型的代码整合到一起,发现有一个模型的代码整合后性能大不如前,但基本上是源码迁移,找了一天原因才发现是因为model.eval()和model.train()放错了位置!!!故在此介 ...
Pytorch：model.train()和model.eval()用法和区别，以及model.eval()和torch.no_grad()的区别
model.train()和model.eval()的区别主要在于Batch Normalization和Dropout两层. model.train() 官方文档启用 Batch Normaliz ...
model.fit以及model.fit_generator区别及用法
model.fit以及model.fit_generator区别及用法_猫爱吃鱼the的博客-CSDN博客
（深入理解）model.eval() 、model.train()以及torch.no_grad() 的区别
文章目录简要版解释深入版解释简要版解释在PyTorch中进行validation或者test的时侯,会使model.eval()切换到测试模式,在该模式下,model.training=Fas ...
【pytorch】model.train和model.eval用法及区别详解
使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval,eval()时,框架会自动把BN和DropOut固定住,不会取平均,而是用训练好的值,不然的话,一旦test的 ...
model.parameters()，model.state_dict()，model .load_state_dict()以及torch.load()
一.model.parameters()与model.state_dict() model.parameters()与model.state_dict()都是Pytorch中用于查看网络参数的方法一 ...
MVC中Model BLL层Model模型互转
MVC中Model BLL层Model模型互转一. 模型通常可以做2种:充血模型和失血模型,一般做法是模型就是模型,不具备方法来操作,只具有属性,这种叫做失血模型(可能不准确):具备对模型一定的简单 ...

Model.modules和Model.children

Model.modules和Model.children相关推荐

最新文章

热门文章