目录： 分类模型训练代码 分类模型测试代码 自定义损失函数 标签平滑 mixup训练 L1正则化 不对偏置项进行权重衰减 梯度裁剪 得到当前学习率 学习率衰减 优化器链式更新 模型训练可视化 保存和加载断点 提取Imagenet预训练模型 ...

一个简洁、好用的Pytorch训练模板 代码地址：https://github.com/KinglittleQ/Pytorch-Template 怎么使用 1) 更改template.py 替换 __init__方法中的内容，增添自己的模型、优化器、评估器等等. 2) 写部分训练代码 ...

https://www.jianshu.com/p/fb132fdbde3b ...

1.仓库地址 https://github.com/meijieru/crnn.pytorch 原版用lua实现的：https://github.com/bgshih/crnn 需要用到的warp_ctc_pytorch： https://github.com/SeanNaren ...

Pytorch多GPU训练 临近放假, 服务器上的GPU好多空闲, 博主顺便研究了一下如何用多卡同时训练 原理 多卡训练的基本过程 首先把模型加载到一个主设备 把模型只读复制到多个设备 把大的batch数据也等分到不同的设备 最后将所有设备计算得到的梯度合并更新 ...

Pytorch预训练模型以及修改 pytorch中自带几种常用的深度学习网络预训练模型，torchvision.models包中包含alexnet、densenet、inception、resnet、squeezenet、vgg等常用网络结构，并且提供了预训练模型，可通过调用来读取网络结构和预 ...

断点续训的效果基本和直接训练一致，但仍有些差别，后面会继续分析 ...

加快Pytorch训练速度 num_workers num_worker=0表示只用主进程读取数据， num_worker=4/8表明使用额外的4/8子进程读取数据 一般来说，增大num_worker可以增加GPU的利用率，从而加快训练的速度。 但是有时候即使增大 ...