activatedgeek / activatedgeek/LeNet-5
CrossEntropyLoss with LogSoftmax?
Aperta
- Lingua principale
- Python
- Stelle
- 252
- Fork
- 101
- Metriche di merge delle PR
- Nessuna PR unita negli ultimi 30g
Descrizione
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() in one single class.
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Valutazione
Questa issue non è ancora stata valutata.