Skip to content

BecauseY/mnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

#1.数据集说明 mnist:https://zhuanlan.zhihu.com/p/547147237?utm_id=0

#2.文件说明:

  • checkpoints:模型文件存储位置
  • data:数据集存储位置
  • model:模型代码
  • train.py: 训练代码
  • predict.py:预测代码

#2.使用流程:

  • 运行train.py进行训练
右键run,会使用默认值运行。

或者命令行传参运行:
python train.py --gpu 0 --epochs 50 --batch_size 64 --lr 0.1 --dataset_path ./data/
  • 运行predict.py进行单个图片的预测
右键run,会使用默认值运行。

或者命令行传参运行:
python predict.py --gpu 0 --model_savepath ./checkpoints/mnist-lenet5.pth --dataset_path ./data/

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages