PytorchExample icon indicating copy to clipboard operation
PytorchExample copied to clipboard

Pytorch最小实践

Pytorch最小实践

收录到PytorchNetHub

AMP

  • 自动混合精度训练

DDP

  • 分布式数据并行(多机多卡)

MNN/NCNN部署

  • Pytorch->ONNX-> NCNN / MNN

TensorRT部署

  • TensorRT API
  • Pytorch->ONNX->TensorRT