This project focuses on implementing basic deep learning models using PyTorch. It includes:
- CNN for image classification (CIFAR-10)
- RNN, LSTM and GRU for sequence learning
- GAN for image generation (Fashion-MNIST)
The main objective is to understand how these models work and observe their performance during training.
deep-learning-project/ │ ├── main.py ├── requirements.txt ├── README.md │ ├── outputs/ │ └── generated_images/
- Install the required libraries:
pip install -r requirements.txt
- Run the program:
python main.py
CNN:
- Trained on CIFAR-10
- Loss decreased over epochs
RNN, LSTM, GRU:
- LSTM and GRU performed better than RNN
GAN:
- Outputs improved over epochs
outputs/generated_images/
- CNN works well for images
- LSTM/GRU better than RNN
- GAN training is unstable
This project helped in understanding CNN, RNN and GAN models practically.
Susan Riona D'Souza