Pytorch Cats and Dogs Recognition
Collection
4 items
•
Updated
Model Card for Cats vs Dogs Classifier
Model Description
This model is a Convolutional Neural Network (CNN) designed to classify images of cats and dogs. It was trained on the "louiecerv/cats_dogs_dataset" dataset.
Intended Use
The model is intended for educational purposes and as a demonstration of image classification using CNNs. It can be used to classify images of cats and dogs.
Training Data
The model was trained on the "louiecerv/cats_dogs_dataset" dataset, which contains images of cats and dogs.
Training Procedure
The model was trained using the following procedure:
- Optimizer: Adam
- Learning Rate: 0.001
- Loss Function: CrossEntropyLoss
- Number of Epochs: 10
- Batch Size: 32
Evaluation
The model was evaluated on the training dataset. Further evaluation on a separate validation or test dataset is recommended.
Limitations
The model may not perform well on images that are significantly different from the training data. It is also limited to binary classification (cats vs. dogs).
License
The model is licensed under the MIT License.
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
HF Inference API was unable to determine this model's library.