Pytorch --- nn.Sequential()模組
簡而言之,nn.Sequential()可以將一系列的操作打包,這些操作可以包括Conv2d()、ReLU()、Maxpool2d()等,打包後方便呼叫吧,就相當於是一個黑箱,forward()時呼叫這個黑箱就行了。
節選AlexNet程式碼的一部分來理解sequential:
class AlexNet(nn.Module):
def __init__(self, num_classes=1000, init_weights=False):
super(AlexNet, self).__init__()
self.features = nn.Sequential(
nn.Conv2d(3, 48, kernel_size=11, stride=4, padding=2),
nn.ReLU(inplace=True),
nn.MaxPool2d(kernel_size=3, stride=2),
nn.Conv2d(48, 128, kernel_size=5, padding=2),
nn.ReLU(inplace=True),
nn.MaxPool2d(kernel_size=3, stride=2),
nn.Conv2d(128, 192, kernel_size=3, padding=1),
nn.ReLU(inplace=True),
nn.Conv2d(192, 192, kernel_size=3, padding=1),
nn.ReLU(inplace=True),
nn.Conv2d(192, 128, kernel_size=3, padding=1),
nn.ReLU(inplace=True),
nn.MaxPool2d(kernel_size=3, stride=2),
)
......
def forward(self, x):
x = self.features(x)
......
return x
__init__中 self.features = nn.Sequential(…)
在forward()中只需要使用self.features(x)就可
相關文章
- pytorch之hub模組PyTorch
- 十五、PyTorch常用工具模組PyTorch
- transforms模組—PyTorch影像處理與資料增強方法ORMPyTorch
- [PyTorch 學習筆記] 2.2 圖片預處理 transforms 模組機制PyTorch筆記ORM
- pytorchPyTorch
- 配置pytorchPyTorch
- Pytorch QuickStartPyTorchUI
- Pytorch | Pytorch格式 .pt .pth .bin .onnx 詳解PyTorch
- 《PyTorch》Part5 PyTorch之遷移學習PyTorch遷移學習
- (pytorch-深度學習系列)pytorch資料操作PyTorch深度學習
- 【Pytorch教程】迅速入門Pytorch深度學習框架PyTorch深度學習框架
- pytorch簡介PyTorch
- PyTorch入門PyTorch
- pytorch 學習PyTorch
- 安裝pytorchPyTorch
- Pytorch - DataloaderPyTorch
- pytorch安裝PyTorch
- 【小白學PyTorch】10 pytorch常見運算詳解PyTorch
- 【小白學PyTorch】13 EfficientNet詳解及PyTorch實現PyTorch
- 【小白學PyTorch】12 SENet詳解及PyTorch實現PyTorchSENet
- pytorch安裝命令PyTorch
- PyTorch-TensorBoardXPyTorchORB
- pytorch 方法筆記PyTorch筆記
- pytorch安裝教程PyTorch
- 編譯 PyTorch 模型編譯PyTorch模型
- Pytorch筆記(一)PyTorch筆記
- Pytorch入門-TransformsPyTorchORM
- Pytorch Dataset入門PyTorch
- pytorch 包介紹PyTorch
- Make Your First GAN With PyTorch 之 第一個 PyTorch 神經網路PyTorch神經網路
- Pytorch系列:(四)IO操作PyTorch
- 【AI】Pytorch_LearningRateAIPyTorch
- Pytorch入門演練PyTorch
- Pytorch的API詳解PyTorchAPI
- pytorch入門(七):unsqueezePyTorch
- pytorch學習筆記PyTorch筆記
- 使用Pytorch搭建模型PyTorch模型
- Pytorch分散式訓練PyTorch分散式