Skip to content

Midterm Project for DL, a ResNet model implementation using the PyTorch framework for image classification tasks.

Notifications You must be signed in to change notification settings

Tejazzz/Implementation-of-ResNet-Model-for-CIFAR-10-Dataset-with-Data-Augmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

Implementation of ResNet Model for CIFAR-10 Dataset with Data Augmentation

Developed a deep learning model using a custom ResNet architecture for image classification on the CIFAR-10 dataset. The model achieved a test accuracy of 91.65% with 4,991,306 trainable parameters. Utilized PyTorch framework, torchvision library, and implemented data augmentation techniques for improving model generalization. Employed early stopping and Adam optimizer during training for better convergence and model performance. Additionally, visualized training and validation accuracy/loss curves for model analysis.

About

Midterm Project for DL, a ResNet model implementation using the PyTorch framework for image classification tasks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published