/pytorch-lesson-zh

pytorch 包教不包会

Primary LanguageJupyter Notebook

pytorch 包教不包会

pytorch-tutorial-zh

雏鹰起飞部分是为了能够快速上手PyTorch

小试牛刀部分则是用来练手的模型,大部分都是对论文工作的复现,或者是一些有意思的例子。

为了避免 jupyter notebook 加载过慢,可以直接选择看 .py 文件,代码和 notebook 中基本一样,只是少了一些图示说明罢了。

如何安装Pytorch?

安装 Pytorch 0.4.0版本,linux、windows 下可以通过这个 Pytorch官网

一、雏鹰起飞

Content .ipynb 文件 .py 文件
1.Tensor基础 Tensor基础.ipynb Tensor基础.py
2.autograd机制 autograd机制.ipynb autograd机制.py
3.线性回归 线性回归.ipynb 线性回归.py
4.多层感知机 多层感知机.ipynb 多层感知机.py
5.Dataset和DataLoader Dataset和DataLoader.ipynb Dataset和DataLoader.py
6.CNN和MNIST CNN和MNIST.ipynb CNN和MNIST.py
7.参数初始化和使用预训练模型 参数初始化和使用预训练模型.ipynb 参数初始化和使用预训练模型.py
8.模型微调的各种trick 模型微调的各种trick.ipynb 模型微调的各种trick.py
9.模型保存和加载 模型保存和加载.ipynb 模型保存和加载.py
10.循环神经网络(RNN) 循环神经网络(RNN).ipynb 循环神经网络(RNN).py

二、小试牛刀

1、计算机视觉——经典算法(卷积神经网络专区)

Content .ipynb 文件 .py 文件 paper
AlexNet AlexNet.ipynb AlexNet.py AlexNet paper
VGG VGG.ipynb VGG.py VGG paper
Network In Network NIN.ipynb NIN.py Network In Network paper
GoogleNet [GoogleNet.ipynb] [GoogleNet] GoogleNet V1 paper
ResNet ResNet.ipynb ResNet.py ResNet paper
DenseNet [DenseNet.ipynb] [DenseNet] DenseNet paper

2、计算机视觉——应用领域

Content .ipynb 文件 .py 文件 paper
语义分割(FCN) FCN.ipynb FCN.py FCN paper

[YOLO.ipynb]

[Image2Language.ipynb]

3、自然语言处理

Content .ipynb 文件 .py 文件 paper
Word2Vec Word2Vec.ipynb Word2Vec.py Word2Vec Toolkit
LSTM 使用LSTM来生成周杰伦歌词.ipynb 使用LSTM来生成周杰伦歌词.py [paper]
Encoder-Decoder 使用Encoder-Decoder来完成机器翻译.ipynb Encoder-Decoder.py [paper]
注意力机制 使用Encoder-Decoder + Attention 机制来完成机器翻译.ipynb Attention.py Neural Machine Translation by Jointly Learning to Align and Translate

[情感分类.ipynb]

[聊天机器人.ipynb]

4、生成模型

Content .ipynb 文件 .py 文件 paper
GAN GAN.ipynb GAN.py GAN paper
DCGAN DCGAN.ipynb DCGAN.py DCGAN paper
Variational Auto-Encoder VAE.ipynb VAE.py VAE paper

[WGAN.ipynb]

5、样式迁移

Content .ipynb 文件 .py 文件 paper
Nueral_Style Neural Style.ipynb Nueral Style.py Nueral_Style paper

6、强化学习

Content .ipynb 文件 .py 文件 paper
Policy Gradients 基于 policy gradients 来玩 CartPole-v0 基于 policy gradients 来玩 CartPole-v0 任务介绍

7、其它

Content .ipynb 文件 .py 文件 paper
Capsule Capsule.ipynb Capsule.py Capsule paper
Attention is all you need attention_all.ipynb attention_all.py Attention is all you need

Dependencies

Python 3.5

PyTorch 0.3.0

最好有 GPU

Reference

PyTorch官方docs 0.2.0版本

PyTorch官方tutorial

DeepNLP-models-Pytorch

PyTorchZeroToAll

MorvanZhou/PyTorch-Tutorial

yunjey/pytorch-tutorial

李沐—gluon教程

吴恩达——Deep Learning教程