pytorch loss 계산 시 type Error 해결

이성범·2022년 4월 26일
0

Development

목록 보기
2/7
  • pytorch는 기본적으로 FloatTensor, LongTensor 등의 초기 type이 각각 torch.float32, torch.long 로 설정되어 있음
  • pytorch는 연산시 데이터의 type이 다르면 loss 계산 등에 오류가 발생함
  • 따라서 초기 input, target 값을 적절한 type으로 설정해주는 것이 중요함

torch.tensor(feature, dtype = torch.long or torch.float32 등)
profile
Machine Learning Engineer at Konan Technology

0개의 댓글