SGD takes "cost" as one of the params. In this case it should be NLL. But all I can find is a Softmax class in MLP.