Skip to content

killerray/dnnBox

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

dnnBox

Learn to implement deep learning networks using PyTorch

学习使用pytorch实现LeNet、AlexNet、LSTM、BiLSTM、CNN-LSTM,使用pytorhch自定义LeNet、AlexNet、BiLSTM模型处理识别MNIST数据集中的手写数字,定义CNN-LSTM处理自定义数据集。

  1. 模型定义:model.py

  2. 训练&验证:train.py

  3. 程序调用:run.py

  4. 自定义数据集加载:dataset模块

About

Learn to implement deep learning networks using PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages