코드를 작성할 때
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)