tqdm(enumerate(dataloader))와 enumerate(tqdm(dataloader))
tqdm은 for문에서 iterable을 순회할때 어느정도 남아있는지 알려주는 대표적인 라이브러리
기본적으로 tqdm(iterable) 형태로 다음과 같이 사용
필요에 따라 dataloader를 순회할때 어느정도 남아있는지 궁금하다면...
dataloader는 indexing이 불가능해서 range(len(dataloader)): 형태로는 사용하기 어렵다
그래서 enumerate(dataloader)로 사용하는데 여기서 tqdm을 어떻게 써야하나
tqdm(enumerate(train_dataloader)) 형태로 쓰면... 이렇게 어느 정도 남아있는지 알수가 없음
반대로 enumerate(tqdm(train_dataloader)) 형태로 사용하면 어느정도 남아있는지 보여줌
'프로그래밍 > Pytorch' 카테고리의 다른 글
NLP text data 전처리에서 tokenizing할 때 padding이 필요한 이유 (0) | 2024.03.31 |
---|---|
NLP huggingface model에 input을 제대로 넣었는데 IndexError: index out of range in self가 나는 이유 (0) | 2024.03.31 |
pytorch의 tensor를 plt.imshow()했더니 TypeError: Invalid shape for image data (0) | 2023.11.07 |
pytorch - flatten과 averaging pooling, training 방법 기본기, layer 구성법 (0) | 2023.05.08 |
pytorch - 모델의 parameter 제대로 이해하기 재활치료 (0) | 2023.05.01 |
TAGS.