nn.CrossEntropyLoss()
-> Softmax+ CrossEntropy
multilayer 분류에서 결과값만 얻길 원할때는 softmax가 필요없음.
model을 만들때 꼭 Softmax를 마지막 레이어에 넣을 필요 X
But
BCE는 Activation function적용 해줘야함!
'AI' 카테고리의 다른 글
| what is backbone? (0) | 2022.10.20 |
|---|
nn.CrossEntropyLoss()
-> Softmax+ CrossEntropy
multilayer 분류에서 결과값만 얻길 원할때는 softmax가 필요없음.
model을 만들때 꼭 Softmax를 마지막 레이어에 넣을 필요 X
But
BCE는 Activation function적용 해줘야함!
| what is backbone? (0) | 2022.10.20 |
|---|