CrossEntropy Loss를 많이 사용하는데 Pytorch의 경우 Class가 2개일 경우(즉, Class가 0, 1) BCELoss 또는 BCEWithLogitsLoss를 사용해야된다.

Pytorch의 CrossEntropy Loss는 Class가 3이상에서만 돌아가기 때문이다. (즉, 마지막 레이어 노드수가 2개 이상이어야함.)