It depends on your actual use case and the applied loss function.
For a multi-class classification problem, you wouldn’t use a softmax layer as your activation output, but pass the raw logits into nn.CrossEntropyLoss (or use F.log_softmax and nn.NLLLoss alternatively).
That being said, I’m still unsure, how the graphs were created and if any conversion library added these softmax layers or where these layers are generally coming from.