Pytorch

 

criterion

loss를 왠일인지 criterion이라고 부른다.

 

예를 들어 binary corss entropy loss를 설정하고 실제 그 값을 계산하고 싶으면

 

 

criterion = nn.BCELoss()  # Binary cross entropy: http://pytorch.org/docs/nn.html#bceloss
decision = D(input)
error = criterion(decision, Variable(torch.zeros(1)))

 

 

위처럼 하면 된다. 네트웍 output 과 label을 넣어주면 loss 계산