keris2020 / hackathon

10 stars 7 forks source link

CrossEntropyLoss함수를 사용하고 싶습니다 #82

Open ghost opened 3 years ago

ghost commented 3 years ago

cre cre_

제가 BCEWithLogitsLoss()함수를 CrossEntropyLoss()함수로 수정했는데, 이런 오류가 뜹니다. CrossEntropyLoss함수를 사용하기 위해 추가적으로 해줘야할 작업이 있나요? ex)softmax 함수 작업 등