tqdm 사용

Minseung Kim·2023년 11월 5일
0
post-thumbnail

tqdm을 사용하며 기존에 알고 있던 방식가 약간 다른 방식을 사용하게 되어 기억에 남기고자 글을 쓰게되었다.


공식 RetinaNet을 학습할 일이 생겨 학습을 진행하며 tqdm으로 학습 진행 상황을 시각화 하려는데 기존에는

images, labels in enumerate(train_load)

와 같은 형태의 코드에서

for images, labels in tqdm(train_load):

와 같은 형태로 변경하여 학습을 진행하면 학습 시 게이지바가 등장하였는데
이번에 진행할 코드는

 for iter_num, data in enumerate(dataloader_train):

로 image와 label을 직접적으로 추출하는 방식이 아닌 코드였다.
위 내용에서

 for iter_num, data in tqdm(dataloader_train,leave=True):

를 사용하니 ValueError: too many values to unpack (expected 2) 오류가 등장하였다.
위 문제를 해결하기 위해 검색을 한 끝에

for iter_num, data in enumerate(tqdm(dataloader_train,leave=True)):

으로 코드를 바꾸는 정상적으로 학습이 잘 진행되었다!!

profile
꾸준히, 열심히, 즐겁게

0개의 댓글