Learn to implement deep learning networks using PyTorch
学习使用pytorch实现LeNet、AlexNet、LSTM、BiLSTM、CNN-LSTM,使用pytorhch自定义LeNet、AlexNet、BiLSTM模型处理识别MNIST数据集中的手写数字,定义CNN-LSTM处理自定义数据集。
-
模型定义:
model.py -
训练&验证:
train.py -
程序调用:
run.py -
自定义数据集加载:
dataset模块