PytorchProjects

🐾 Animal Prediction Using PyTorch

This project demonstrates a custom image classification model built using PyTorch. It uses a Convolutional Neural Network (CNN) to classify images of animals into predefined categories.

📌 Features

• Custom CNN built using PyTorch
• Model training and validation with accuracy tracking
• Optimizer: Adam
• Techniques used: Batch Normalization
• Easily extendable to new datasets or classes