activatedgeek / activatedgeek/LeNet-5
CrossEntropyLoss with LogSoftmax?
Offen
- Vorherrschende Sprache
- Python
- Sterne
- 252
- Forks
- 101
- PR-Merge-Kennzahlen
- Keine gemergten PRs in 30 T.
Beschreibung
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.
Beitragsleitfaden
Für dieses Repository ist kein Beitragsleitfaden indexiert
Bewertung
Dieses Issue wurde noch nicht bewertet.