Update README.md

README.md

@@ -12,12 +12,8 @@ tags:
 
 # Model Card for Model ID
 
-
-
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/624d888b0ce29222ad64c3d6/X7cXpayiKgUCUycIen22S.png)
 
-
-
 ## Model Details
 
 ### Model Description
@@ -32,25 +28,32 @@ of 3926 images and an unlabeled dataset of 4000 test images.
 - **License:** [More Information Needed]
 - **Finetuned from model [optional]:** [More Information Needed]
 
-
 ## Training Details
 
 ### Training Data
 
-The training data consists of an unkown subset of the cub-200-2011 dataset, https://paperswithcode.com/dataset/cub-200-2011
+The training data consists of an unknown subset of the cub-200-2011 dataset, https://paperswithcode.com/dataset/cub-200-2011
 
 ### Training Procedure
 
 <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
 
-#### Preprocessing [optional]
-
-[More Information Needed]
+#### Preprocessing
 
+Data augmentation was applied to the training data in a custom Torch dataset class. Because of the size of the dataset images were not replaced but duplicated and augmented.
+The only augmentations applied were HorizontalFlips and Rotations (10 degrees) to align with the relatively homogenous dataset.
 
 #### Training Hyperparameters
 
-- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+| Hyperparameter        | Value                      |
+|-----------------------|----------------------------|
+| Optimizer             | AdamW                     |
+| Learning Rate         | 1e-4                     |
+| Batch Size            | 32                        |
+| Epochs                | 2                        |
+| Weight Decay          | -                      |
+| Scheduler             | -         |
+| Mixed Precision       | Torch AMP                      |
 
 #### Speeds, Sizes, Times [optional]
 
@@ -66,7 +69,7 @@ The training data consists of an unkown subset of the cub-200-2011 dataset, http
 
 #### Testing Data
 
-<!-- This should link to a Dataset Card if possible. -->
+The testing data consists of an unknown subset of the cub-200-2011 dataset, https://paperswithcode.com/dataset/cub-200-2011
 
 [More Information Needed]
 
@@ -78,4 +81,5 @@ The training data consists of an unkown subset of the cub-200-2011 dataset, http
 
 #### Metrics
 
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+