/Loss-Function-In-PyTorch

Loss Function in PyTorch

Primary LanguageJupyter NotebookMIT LicenseMIT

Loss-Function-In-PyTorch

Loss Function in PyTorch


  1. L1范数损失 L1Loss
  2. 均方误差损失 MSELoss
  3. 交叉熵损失 CrossEntropyLoss
  4. KL 散度损失 KLDivLoss
  5. 二进制交叉熵损失 BCELoss
  6. BCEWithLogitsLoss
  7. MarginRankingLoss
  8. HingeEmbeddingLoss
  9. 多标签分类损失 MultiLabelMarginLoss
  10. 平滑版L1损失 SmoothL1Loss
  11. 2分类的logistic损失 SoftMarginLoss
  12. 多标签 one-versus-all 损失 MultiLabelSoftMarginLoss
  13. cosine 损失 CosineEmbeddingLoss
  14. 多类别分类的hinge损失 MultiMarginLoss
  15. 三元组损失 TripletMarginLoss
  16. 连接时序分类损失 CTCLoss
  17. 负对数似然损失 NLLLoss
  18. NLLLoss2d
  19. PoissonNLLLoss