토니의 연습장
Conv2d / ConvTranspose2d 본문
# For nn.Conv2d: (reduce the size of the image for prediction : Discriminator)
output_size = (n + 2 * pad - ks) // stride + 1
# For nn.ConvTranspose2d: (grow from a value to full image for generation : Generator)
output_size = (n - 1) * stride - 2 * padding + ks
'AI 일반 > 모델, 아키텍처, 구현' 카테고리의 다른 글
loss 값 저장 및 출력 (5) | 2025.01.14 |
---|---|
출력 정밀도 변환 클래스 정의 (CastOutputToFloat) (5) | 2024.11.14 |
Fine Tuning (Transfer Learning의 한 종류) (1) | 2024.08.28 |
Feature Extraction (Transfer Learning의 한 종류) (0) | 2024.08.28 |
Underfitting/Overfitting (0) | 2024.08.28 |