Effective Pytorch
Contact me
- Blog -> https://cugtyt.github.io/blog/index
- Email -> cugtyt@qq.com
- GitHub -> Cugtyt@GitHub
高效使用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