# batch 개수
len(dataloader)
# train/test data 개수
len(dataloader.dataset)
'Code > Pytorch' 카테고리의 다른 글
[Pytorch, CUDA] Pytorch 와 CUDA version mismatch (0) | 2025.03.02 |
---|---|
[Pytorch, timm] Optimizer & Parameter Group Learning Rate (0) | 2023.06.27 |
[Pytorch] Seed 고정 (0) | 2023.05.09 |
[Pytorch] GPU에 맞는 CUDA version 설치 (0) | 2023.04.14 |
[Pytorch] pytorch version 확인 (0) | 2023.04.14 |