[오류해결] Dataloader collate function 뻘짓기록

\lib\site-packages\torch\utils\data_utils[collate.py](http://collate.py/)”, line 141, in default_collate return torch.stack(batch, 0, out=out) RuntimeError: stack expects each tensor to be equal size, but got [1, 5985] at entry 0 and [234, 5985] at entry 1

위의 코드를 마주했다면.. loader을 불러오는 부분을 꼭 확인하기..

custom한 collate wrapper을 적용해줬을 가능성이 농후함,,, custom하게 collate function을 지정해줬으면서.. 설정해주는 부분을 까먹었을 수 있으니까.. 아래와 같이.. 설정 꼭 해주길..

self.train_loader = DataLoader(train_dataset, batch_size=self.args.batch_size, shuffle=True, num_workers=self.args.num_workers, collate_fn=train_dataset.collate_wrapper)