TypeError: tuple indices must be integers or slices not str

Pear_Mh·2022년 8월 3일
0
  • Dataset에서 Transform 을 했을 때, Totensor를 적용하였으나, 위의 문제가 발생했었다.
  • 머리를 싸매고 code by code로 확인한 결과, Collate_fn적용에서 문제가 발생한 것을 확인했다.
  • Collate_fn 을 사용하는 이유! map-style 데이터셋에서 sample list를 batch 단위로 바꾸기 위해 필요한 기능입니다. zero-padding이나 Variable Size 데이터 등 데이터 사이즈를 맞추기 위해 많이 사용합니다.
  • 본인이 작성한 Collate_fn의 경우 아래와 같다.
    def collate_fn(batch):
        return tuple(zip(*batch))
  • 이렇게 했다 보니 Dataloader 적용 후 데이터 format이 tuple로 되어있기에 위의 문제가 발생함을 확인했다.
  • 이는 CE 에 대한 기본 개념을 명확히 하지 않은 것으로 판단하여 더욱 공부해야할 것 같다..
profile
Beyond the new era.

0개의 댓글