Pytorch实现常用代码笔记

Pytorch实现常用代码笔记

• • 基础实现代码
  • 其他代码示例
  • • [Network Modules](#Network Modules)
    • Loss
    • Utils

基础实现代码

参考 深度学习手写代码

其他代码示例

Network Modules

Pytorch实现Transformer代码示例

Loss

PyTorch实现CrossEntropyLoss示例
Focal Loss 原理详解及 PyTorch 代码实现
PyTorch实现三元组损失Triplet Loss

Utils

PyTorch实现NMS算法
目标检测中的IoU损失函数