Skip to content

Glossary: Computer Vision Basics

CrossEntropyLoss

Also written: nn.CrossEntropyLoss

The usual loss for picking one class out of several. It compares the logits with the class index, and its optional `weight` gives each class its own weight.

Where the book teaches it

Related terms

logitBCEWithLogitsLoss