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
'알쓸신잡' 카테고리의 다른 글
Ubuntu에서 Filezilla3 설치 (0) | 2024.08.11 |
---|---|
dlopen(): error loading libfuse.so.2 에러 해결 (0) | 2024.08.11 |
MeshLab 설치 및 에러 해결 (1) | 2024.05.09 |
학습 시 두 개 이상의 데이터셋에서 batch 뽑아내는 법 (0) | 2023.04.08 |
티스토리(tistory) 모바일에서 LaTex 수식 깨짐 해결 (4) | 2023.03.26 |