The vocabulary of neural networks — the neuron and its parameters, the activation functions, and the forward and backward passes that train it — each with a small picture.
CrossEntropyLoss and BCEWithLogitsLoss expect logits, not probabilities — applying softmax yourself first is a classic bug.CrossEntropyLoss applies softmax itself — feed it logits, not probabilities.