Masked Loss 구현

이성범·2022년 4월 26일
0

Development

목록 보기
3/7
  • 우리는 가끔 목적에 맞게 모델을 학습시키기 위해 Loss를 계산할 때 원하는 위치의 값들만 사용할 때가 있음
  • 이럴때 필요한 방법이 Masked Loss 임
  • 아래는 간단한 Masked Loss 구현 Code이고, 아래에 인사이트를 이용하여 상황에 맞게 Masked Loss를 구현하면됨
# target.shape == output.shape
criterion = torch objective function

criterion(output[target == 원하는 값], target[target == 원하는 값])
profile
Machine Learning Engineer at Konan Technology

0개의 댓글