This project implements a cloth prediction model using PyTorch, trained on the FashionMNIST dataset. The model is designed to classify different types of clothing items with high accuracy.
- Utilizes PyTorch for building and training the machine learning model.
- Achieves accurate classification of clothing items from the FashionMNIST dataset.
- Provides a straightforward example of implementing a deep learning model for image classification.
- Python: The project is written in Python.
- PyTorch: PyTorch is used for building and training the machine learning model.
- FashionMNIST Dataset: The model is trained on the FashionMNIST dataset.
- Clone the repository:
git clone https://github.com/amith7025/cloth-prediction-model.git cd cloth-prediction-pytorch