부캠 TIL 0824 - PyTorch Dataset, DataLoader

기린이·2021년 8월 24일
0

부캠 TIL 🦆

목록 보기
17/53
post-thumbnail

질문

  • torchvision.transform의 augmentation할때 totensor의 위치가 다다른데 언제로 해야하는지?
    document에 답이 있다. 인풋으로 tensor,PIL을 받는 함수들은 앞에서 totensor를 해줘도 상관이 없을 것이다.

Dataset

  • 얼굴의 위치의 분포를 확인해서 crop
  • 필터를 넣어서 APTOS예시처럼 분별이 쉽도록 도와줄수도 있다.

Generalization

  • validatation set
  • data augmentation - torchvision.transform, albumentation

Data Generation

data를 만들어내는 속도를 체크하는 것이 중요할때가 있다.

torch.utils.data

Dataset

DataLoader

  • drop_last = True
  • sampler와 shuffle의 관계
  • collate_fn : 배치마다 적용하고 싶은 함수

그외

wandb사용해서 체킹해보자!!

profile
중요한 것은 속력이 아니라 방향성, 공부하며 메모를 남기는 공간입니다.

0개의 댓글