반응형 연구 도구도구/Tools1 [연구노트] DataParallel 공부하는 중 (feat. panoptic deeplab...) 버그. 요즘 Panoptic deeplab을 reproducing하고있다. 현재 에러가 나는 이유가, 여기선 데이터로더로 배치를 불러올 때 단순히 이미지 tensor만 가져오지 않아서이다. dictionary의 리스트로 되어있다! 그러면 만약에 16배치면, 리스트 안에 16개의 딕셔너리가 있는거다. [{}, {}, {}, {}, ... , {}] 그리고 딕셔너리 안에 이미지 텐서 (그외 필요한 각종 데이터)가 들어있는 형태. 근데 그러면 리스트의 딕셔너리에서 꺼내서 전부 합쳐서 데이터로 넣어야한다. 이미지만 그렇게 하는건 할 수 있는데, 또 모델 안에서 이미지 외에 다른 데이터도 필요로하는 바람에 뭔가 자꾸 에러가 뜬다. 뇌빼고 코딩하면 그냥 리스트를 없애고 딕셔너리를 전부 합쳐버리면 되긴 하는데, 코드.. 2021. 2. 8. 이전 1 다음 반응형