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.
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.
1import torch
2from model import ImageClassifier
3
4model = ImageClassifier(input_size=128*128*3, n_classes=2)
5model.load_state_dict(torch.load("cats_dogs_classifier.pth"))
6model.eval()
This model is released under the MIT license.
