Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | ||
6 | 7 | 8 | 9 | 10 | 11 | 12 |
13 | 14 | 15 | 16 | 17 | 18 | 19 |
20 | 21 | 22 | 23 | 24 | 25 | 26 |
27 | 28 | 29 | 30 | 31 |
Tags
- pytorch
- 개인 PT
- 개인 피티
- 바디프로필
- 체스트프레스
- 코드
- 데드리프트
- 하체운동
- 코테 공부
- 운동
- 바프준비
- 연구 시작
- 다이어트
- 영화 비평
- github
- Knowledge Tracing
- 오블완
- 티스토리챌린지
- 프로그래머스
- 개발자
- 코딩테스트
- 암풀다운
- 건강
- 디버깅
- 덤벨운동
- 라섹 수술 후기
- 코테준비
- PT 운동
- 논문 리뷰
- 개인 운동
Archives
- Today
- Total
치즈의 AI 녹이기
[pytorch] collate_fn에 arg 추가하기 본문
data.DataLoader에서 사용하는 collate_fn은
일반적으로 사용자 정의 함수에 의한 batch 단위의 데이터를 처리를 할 때 이용한다.
따라서 사용자 정의 함수의 argument는 batch 단위의 데이터 하나만 받는데
나의 경우, 특정 조건에 따라 전처리를 다르게 하기 위해 추가 인자를 넣어줘야 하는 상황이었다.
해결 방법은 간단하게 collate_fn을 위한 새로운 클래스를 생성하여, 추가 인자를 넣어주면 되었다.
# 예시코드
class MyCollator(object):
def __init__(self, *params):
self.params = params
def __call__(self, batch):
# do something with batch and self.params
.
.
.
# feeding to the dataloader
my_collator = MyCollator(param1, param2, ...)
data_loader = torch.utils.data.DataLoader(..., collate_fn=my_collator)
참고 링크: https://intrepidgeeks.com/tutorial/add-parameter-to-torch-collate-fn
[torch] collate_fn에 arguments 추가하기
pytorch Dataloader의 collate_fn 매개변수를 조작하면서 얻은 간단한 해결법에 대해 적는다. pytorch는 torch.utils.data.Dataset과 torch.utils.data.DataLoader의 두 가지 도구를 제공한다. Dataset은 input feature x와 label y
intrepidgeeks.com
'인공지능 대학원생의 생활 > 구글링' 카테고리의 다른 글
서버에서 서버로 파일 옮기는 방법 (0) | 2022.05.27 |
---|---|
Significance Test 하기 (0) | 2022.04.20 |
터미널에서 permission denied 해결하기 (0) | 2022.04.08 |
How to Detach specific components in the loss? (0) | 2022.04.03 |
GAN 이해하기 (0) | 2022.03.22 |