Skip to content

jwkanggist/tf-code-pattern-lenet5

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Readme for TF Code Pattern Design with Lenet5-Mnist

  • Final update: 2018 Oct
  • All right reserved @ Jaewook Kang 2018

About

The aim of this repository is to introduce an exemplary TF project in practice. We show the example with respect to MNIST classification by Lenet5. This project is composed of several .py files, each of which undertakes single role and responsibility according to the OOP philosophy.

- data_loader.py    : Preparing and feeding the dataset in batchwise by using tf.data
- model_builder.py  : Building a model in tensorflow computational graph.
- model_config.py   : Specifying a configulation for the model 
- trainer.py        : Training the model by importing the dataloader and the model_builer
- train_config.py   : Including a configulation for the training
- eval.py           : Evaluating the model with respect to test dataset by loading a ckpt

Model Table

alt text

How to Run

Training

python ./tf_module/trainer.py

Inference

python ./tfmodule/eval.py

Components

./tfmodule/
├── data
│   └── mnist
├── export
│   └── tf_logs
├── data_loader.py
├── eval.py
├── model_builder.py
├── model_config.py
├── testcodes
│   └── test_dataloader.py
├── train_config.py
└── trainer.py

Compiler/Interface Dependencies

  • Tensorflow >=1.9
  • Python2 <= 2.7.12
  • Python3 <= 3.6.0

Related Materials

Feedback

  • Issues: report issues, bugs, and request new features
  • Pull request
  • Email: jwkang10@gmail.com

License

  • Apach License 2.0

Authors information

  • Jaewook Kang Ph.D.
  • Personal website: link
  • Facebook : link
  • Linkedin : link

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages