-
Notifications
You must be signed in to change notification settings - Fork 99
Open
Description
Line 37 in b9d7f39
| ('sig7', nn.LogSoftmax(dim=-1)) |
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.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels