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/