ValueError: too many values to unpack (expected 2) [tqdm 사용]

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을 직접적으로 추출하는 방식이 아닌 코드였습니다.

해당 코드에 기존과 같은 방식을 적용하니 

 

ValueError: too many values to unpack (expected 2)

위와 같은 오류가 발생했습니다.

문제 해결을 위해 검색한 결과 

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

다음과 같이 코드를 적용하니 정상적으로 학습이 잘 진행되었습니다!!