Let's structure our project as shown in the following pattern:
- hy_param.py: All the hyperparameters and other configurations are defined here
- model.py: Definition and architecture of the model are defined here
- train.py: Code to train the model is written here
- inference.py: The code to execute the trained model and make predictions are defined here
- /runs: This folder will store all the checkpoints which get stored during the training process
You can clone the code from the repository and find the code in Chapter 2 folder:
GITHUB LINK: https://github.com/PacktPublishing/Python-Deep-Learning-Projects/