Downloads · 30 days
0
cats_dogs_recognition_torch_nn
louiecerv/cats_dogs_recognition_torch_nn
cats_dogs_recognition_torch_nn is a image classification model from louiecerv. Use it when you need a label for an image. It is set up for torch. The card lists the license as mit.
This model is a simple neural network trained to classify images of cats and dogs. It is built using PyTorch and trained on the louiecerv/catsdogsdataset dataset.
Access
Public
Updated Feb 4, 2025
Repo size
102 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth102 MB · 100%
At a glance
- Task
- Image Classification
- Library
- torch
- License
- mit
- Access
- Public
- Created
- Feb 3, 2025
- Updated
- Feb 4, 2025
- SHA
- 1ff594f6
Model card
From the Hugging Face model README
Cats & Dogs Classifier
This model is a simple neural network trained to classify images of cats and dogs. It is built using PyTorch and trained on the louiecerv/cats_dogs_dataset dataset.
Model Details
- Architecture: Fully connected neural network
- Input Size: 128x128 RGB images
- Number of Classes: 2 (Cats & Dogs)
- Optimizer: Adam
- Loss Function: CrossEntropyLoss
- Training Epochs: 5
- Dataset: Cats & Dogs Dataset
Training
The model was trained using a basic fully connected neural network with ReLU activation functions. The training process involved using the Adam optimizer with a learning rate of 0.001.
Usage
import torch
from model import ImageClassifier
model = ImageClassifier(input_size=128*128*3, n_classes=2)
model.load_state_dict(torch.load("cats_dogs_classifier.pth"))
model.eval()
License
This model is released under the MIT license.
- Task
- Image Classification
- Library
- torch
- License
- mit
- Created
- Feb 3, 2025
- Updated
- Feb 4, 2025