reoneo97 / pytorch-project-template

Production-ready template for PyTorch Deep Learning Projects

pytorch-project-template

Production-ready template for PyTorch Deep Learning Projects
Check out the associated Medium Article: https://towardsdatascience.com/leveling-up-your-machine-learning-projects-317da9c787fe

Project Structure

|-- project
|   |-- __init__.py
|   |-- config.py
|   |-- datasets
|   |   |-- __init__.py
|   |   `-- dataset.py
|   |-- experiment.py
|   `-- models
|       |-- __init__.py
|       `-- conv.py
|-- requirements.txt
|-- README.md
|-- config.yml
`-- main.py