PyTorch: torch.nn.Sequantial

danbibibi·2022년 1월 26일
0

PyTorch 🔥

목록 보기
6/20

torch.nn.Sequantial

코드를 작성할 때 torch.nn.Sequantial을 이용하면 forward() 함수에서 구현될 순전파를 Layer 형태로 가독성 높게 작성할 수 있다.

사용법

import torch
from torch import nn

torch.nn.Sequential(*args)
# example
import torch
import torch.nn as nn

class model(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv = nn.Sequential(
            nn.Conv2d(1, 20, 5),
            nn.ReLU(),
            nn.Conv2d(20, 64, 5),
            nn.ReLU()
        )

    def forward(self, x):
        out = self.conv(x)
profile
블로그 이전) https://danbibibi.tistory.com

0개의 댓글