criterions.rst 758 B

12345678910111213141516171819202122232425262728293031
  1. .. role:: hidden
  2. :class: hidden-section
  3. .. _Criterions:
  4. Criterions
  5. ==========
  6. Criterions compute the loss function given the model and batch, roughly::
  7. loss = criterion(model, batch)
  8. .. automodule:: fairseq.criterions
  9. :members:
  10. .. autoclass:: fairseq.criterions.FairseqCriterion
  11. :members:
  12. :undoc-members:
  13. .. autoclass:: fairseq.criterions.adaptive_loss.AdaptiveLoss
  14. :members:
  15. :undoc-members:
  16. .. autoclass:: fairseq.criterions.composite_loss.CompositeLoss
  17. :members:
  18. :undoc-members:
  19. .. autoclass:: fairseq.criterions.cross_entropy.CrossEntropyCriterion
  20. :members:
  21. :undoc-members:
  22. .. autoclass:: fairseq.criterions.label_smoothed_cross_entropy.LabelSmoothedCrossEntropyCriterion
  23. :members:
  24. :undoc-members: