https://pytorch.org/docs/stable/generated/torch.nn.AdaptiveAvgPool1d.html
AdaptiveAvgPool1d — PyTorch 1.10 documentation
Shortcuts
pytorch.org
이미지 사이즈를 torch.nn.AdaptiveAvgPool1d(output_size) 의 output_size로 맞춘다. 물론 여기선 1d인거고, 2d, 3d도 마찬가지일거임.
'기술 > PyTorch' 카테고리의 다른 글
| DistributedDataParallel 관련 링크 정리 (0) | 2021.12.27 |
|---|---|
| torchvision.models 의 IntermediateLayerGetter (1) | 2021.10.06 |
| pbar 고급진 사용법 (0) | 2021.09.13 |
| nn.Sequential 에 OrderedDict 넣어줘도 됨 (0) | 2021.08.18 |
| 실제 torch weights 살펴보기 (0) | 2021.08.17 |