Update README.md

README.md

@@ -6,6 +6,7 @@ tags:
 model-index:
 - name: resnet18-catdog-classifier
   results: []
+pipeline_tag: image-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -13,11 +14,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # resnet18-catdog-classifier
 
-This model is a fine-tuned version of [microsoft/resnet-18](https://huggingface.co/microsoft/resnet-18) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/resnet-18](https://huggingface.co/microsoft/resnet-18) on an [custom](https://www.kaggle.com/datasets/samuelcortinhas/cats-and-dogs-image-classification) dataset.
 
 ## Model description
 
-More information needed
+This model was built using the "Cats & Dogs Classification" dataset obtained from Kaggle. During the model building process, this was done using the Pytorch framework with pre-trained Resnet-18. The method used during the process of building this classification model is fine-tuning with the dataset.
 
 ## Intended uses & limitations
 
@@ -32,17 +33,15 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 5e-05
-- train_batch_size: 8
-- eval_batch_size: 8
-- seed: 42
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
-- num_epochs: 3.0
+- loss_function = CrossEntropyLoss
+- optimizer = AdamW
+- learning_rate: 0.0001
+- batch_size: 16
+- num_epochs: 10
 
 ### Framework versions
 
 - Transformers 4.33.2
 - Pytorch 2.0.1+cu118
 - Datasets 2.14.5
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3