Code & Framework/Pytorch
[Pytorch] Batch 개수, Data 개수
이성훈 Ethan
2023. 5. 14. 21:11
728x90
# batch 개수
len(dataloader)
# train/test data 개수
len(dataloader.dataset)
728x90