SeongBeomLEE
로그인
SeongBeomLEE
로그인
pytorch loss 계산 시 type Error 해결
이성범
·
2022년 4월 26일
팔로우
0
PyTorch
code
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 등
)
이성범
Machine Learning Engineer at Konan Technology
팔로우
이전 포스트
torch.nn.Embedding 값 설정
다음 포스트
Masked Loss 구현
0개의 댓글
댓글 작성