공식 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)):
으로 코드를 바꾸는 정상적으로 학습이 잘 진행되었다!!