Quick Starter Code with different specifications stitched together in tensorflow 2.0 .
Starter code with best practices for different Deep Learning problems in one command.
This Package provides Tensorflow 2.0 boilerplate code , for deep learning problem domains.
To install, in command line enter -
pip install tf-stitch
View Pypi project on https://pypi.org/project/tf-stitch/
Why use this?
If you have been working in deep learning or have just started, you will find yourself writing the same imports, same training loop, same data fetching commands, same model declaration every time.
Personally, found myself copy-pasting MNIST models to get started with a project. And Even copy-pasting does not work as there are many parts of deep learning code that are needed in different combinations for different problems.
This Package is created just to solve this problem for our most beloved framework Tensorflow 2.0. Deep learning is an exciting field full of awesome ideas and new concepts popping every day. Custom stitched boiler code lets you focus on the uniqueness of your implementation and not the repetitive code.
Keras and Tensorflow 2.0, because of simple API, allows you to implement many ideas in less time. This Package just aims to double(2X) that efficiency.
usage: tf-stitch output_file [-d DOMAIN]
[--dataset DATASET] [--model MODEL] [--training TRAINING]
tf-stitch output_script.py -d=vision
tf-stitch output_notebook.ipynb -d=nlp
output_notebook.ipynb are the generated output files with starter code.
output_file is required for command to run.
File can have extension .py or .ipynb for generating python script or jupyter notebook.
-d , --domain DOMAIN
Domain problem type for selecting appropriate model
and dataset. See
tf_stitch/template.json for default datasets and model for each domain option.
Current domain options are -
Either provide domain argument or dataset and model arguments
- dataset DATASET : Select any of Tensorflow datasets. See catalog.
--model MODEL : Select a Deep Learning Model Current available options -
- conv - Convolution model
- rnn - GRU sequence model
- dnn - Fully connected network
- custom - Custom layer network
--training TRAINING : Type of training loop. Options -
--testing TESTING : If to include testing, True or False
tf-stitch output.ipynb --dataset = cifar100 --model = conv
--training = custom --testing = True
All the templates are in their respective folder in
tf_stitch/templates . Feel free to change templates according to your taste and also submit a pull request if it can be helpful for others as well.
Hey There, this project requires your feedback and contribution to improve. You can contribute, but not restricted to, in following ways -
- Providing Feedback and submitting feature request.
- Apply best and latest TF2.0 practices.
- Add more template and domain options.
Looking forward to hear from you : )
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for tf_stitch-1.2.1-py3-none-any.whl