pear_min.log
로그인
pear_min.log
로그인
TypeError: tuple indices must be integers or slices not str
Pear_Mh
·
2022년 8월 3일
팔로우
0
PyTorch
TROUBLESHOOTING
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 에 대한 기본 개념을 명확히 하지 않은 것으로 판단하여 더욱 공부해야할 것 같다..
Pear_Mh
Beyond the new era.
팔로우
이전 포스트
Error: System limit for number of file watchers reached
다음 포스트
TypeError: tuple indices must be integers or slices not str
0개의 댓글
댓글 작성