DataSet의 getitem이 매번 다른 길이(shape)의 텐서를 리턴하는 경우, batch_size를 2 이상으로 주기 위해서는 collate_fn을 짜서 넣어주어야 한다.
shuffle, batch 외의 다양한 feeding 방법을 사용해야 할 때가 있음. 다이나믹하게 학습 데이터가 변경될 때도.
당연히 dataset 안에 합칠 수 있지만 적절히 분리하여 만들어 둔 것!
저는 서영준 입니다.
DataSet의 getitem이 매번 다른 길이(shape)의 텐서를 리턴하는 경우, batch_size를 2 이상으로 주기 위해서는 collate_fn을 짜서 넣어주어야 한다.
shuffle, batch 외의 다양한 feeding 방법을 사용해야 할 때가 있음. 다이나믹하게 학습 데이터가 변경될 때도.
당연히 dataset 안에 합칠 수 있지만 적절히 분리하여 만들어 둔 것!