WenWeiKun

icon indicating a website link http://frusciante.github.io

icon location I'm a student pursuing a degree in Computer Science. I use GitHub to showcase my coding projects, collaborate with classmates, and learn from other developers.

Results 1 issues of WenWeiKun

CrossEntropyLoss with LogSoftmax?

1 comment

https://github.com/activatedgeek/LeNet-5/blob/b9d7f39a752d23a7de60ea6d679812c401fab57e/lenet.py#L37 I think there is no need to put LogSoftmax layer at the end of network if you are going to use CrossEntropyLoss ? Because CrossEntropyLoss combines nn.LogSoftmax() and nn.NLLLoss()...