728x90
torch.nn.Conv2d(in_channels,
out_channels,
kernel_size,
stride=1,
padding=0,
dilation=1,
groups=1,
bias=True,
padding_mode='zeros')
is_channel : 이미지의 채널 수 (ex, color dim)
out_channel : Filter의 개수 = Output의 depth
kernel_size : Filter의 크기
padding : Convolution이 진행하면서 겹쳐지는 공간을 0으로 채워 연산 수를 동일하게 해준다.
stride : 필터 진행 간격
반응형
'👾 Deep Learning' 카테고리의 다른 글
데이터 아카데미 8일차 수업 정리 (0) | 2021.07.07 |
---|---|
[PyTorch] 기존 제공 model 불러오기 (0) | 2021.04.07 |
GTX 1660 super에 맞는 tensorflow, python, CUDA, Cudnn 버전 (4) | 2021.03.28 |
Optimizer ( Adam, SGD ) (0) | 2021.03.27 |
손실 함수 (loss function) (0) | 2021.03.07 |