알쓸신잡

두 개의 dataloader에서 에서 tqdm 쓰는 법

재바기 2023. 4. 8. 18:56
728x90

tqdm을 사용하기 위해선 tqdm의 인자로 들어오는 게 __len__ 메소드를 가져야 한다.

 

따라서, tqdm안에 아래와 같이

for i, data in tqdm(enumerate(zip(train_dataloader1, train_dataloader2)))

이렇게 쓰면 안되고,

__len__을 가지는 train_dataloader1이나 train_dataloader2 중 하나를 tqdm으로 감싼다.(어차피 둘의 크기는 같을 것이므로)

for i, data in enumerate(zip(tqdm(train_dataloader1), train_dataloader2)):

이렇게 해주면 정상적으로 작동한다.

728x90