Skip to the content.

Effective Pytorch

Contact me


高效使用Pytorch等深度学习相关工具。


卷积网络可视化

pytorch-cnn-visualizations,flashtorch


图像数据增强工具

imgaug, Augmentor


自定义C++运算扩展TorchScript

结合C++代码和TorchScript的基本流程


扩展torch.nn和torch.augograd

实现Function和Module


使用TorchScript部署模型

使用tracing和scripting部署seq2seq模型


使用numpy和scipy扩展Pytorch

使用numpy和scipy写Function


TorchScript的使用

TorchScript的作用和用法


imgaug图像增强

可以用于增强分类,目标检测,分割任务


DALI,加速视觉类数据读取

DALI可以借助GPU提升视觉类数据的解码速度,可以替代深度学习框架中的数据加载模块


pytorch-lightning工具

简化Pytorch代码结构,模块化模型训练,轻松处理复杂设置


分布式数据并行

分布式数据并行的配置,以及如何和模型并行结合


模型并行最佳实践

将不同的模块放到不同的GPU上,可以通过划分输入数据进行加速


保证可复现

设定随机数种子,cudnn.deterministic = True, cudnn.benchmark = False


用C++加载Pytorch模型

用C++加载Pytorch模型


使用TensorBoard

使用TensorBoard做可视化分析


控制自动求导autograd

requires_grad,torch.no_grad(),model.eval()


更快的处理图像

fastai快速处理图像


混合精度训练

apex,fastai,pytorch_lightning混合精度训练


data_prefetcher提高数据加载速度

data_prefetcher提高数据加载速度,来自apex


初始化

模型参数初始化


保存和加载模型

正确load和save


THOP: PyTorch-OpCounter

统计Pytorch模型的MAC和FLOP