The code contains custom models and module for training CNN with pytorch

Here is the REPO for that

File Structure

• models -> For importing model architectures
• train.py -> Contains training loop
• test.py -> Contains code for running model on the test set
• utils.py -> Contains supportive functions
• main.py -> Contains code for fitting models in training and testing loops
• gradcam_utils.py -> Contains code for running gradcam on model