Instructions to use amehta633/cats-dogs-model-470 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use amehta633/cats-dogs-model-470 with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://amehta633/cats-dogs-model-470") - Notebooks
- Google Colab
- Kaggle
Model description
It's pretty simple, the model tries to differentiate between dogs/cats.
Intended uses & limitations
Only JPGs can be used and they are auto resized. Meant for dogs and cats only.
Training and evaluation data
Combined training data from multiple sources. Downloaded around 3000 cat and dog photos from r/dogs and r/cats. Curated kaggle datasets into one large dataset.
Training procedure
Trained using an early stopping for overfitting
Training hyperparameters
The following hyperparameters were used during training:
- optimizer: {'name': 'Adam', 'learning_rate': 0.001, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
- training_precision: float32
- Downloads last month
- 14