Base deep learning template for my mini-project
- Python >= 3.5
- Pytorch >= 1.7.0
- Ignite
- Weight & Bias
mini-template
├── config.py - holds configuration for model, training and evaluation
├── main
│ ├── __init__.py
│ ├── api.py - endpoints api of a trained model
│ ├── dataset.py - custom dataset
│ ├── main.py - entry point for training, evaluation
│ ├── model.py - custom model
│ └── utils.py - utitlites for model, training and evaluation
├── requirements.txt
└── tests - testing code folder for functions