DeepNLP-models-Pytorch
DeepNLP-models-Pytorch copied to clipboard
error in Window-Classifier-for-NER model
In the 4th jupyter file named Word Window Classification and Neural Networks, I found something wrong. Specifically, in the class WindowClassifier, you has used self.softmax = nn.LogSoftmax(dim=1)in the output layer, then you don't need use CrossEntropyLoss() loss, you must use torch.nn.NLLLoss loss