알쓸신잡
두 개의 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