Update README.md

README.md

@@ -14,22 +14,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # paligemma-architecture
 
-This model is a fine-tuned version of [google/paligemma2-3b-pt-448](https://huggingface.co/google/paligemma2-3b-pt-448) on the None dataset.
+This model is a fine-tuned version of [google/paligemma2-3b-pt-448](https://huggingface.co/google/paligemma2-3b-pt-448) on a custom architecture dataset.
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
 
 ## Training procedure
 
+Followed the [notebook from smol-vision](https://github.com/merveenoyan/smol-vision/blob/main/Fine_tune_PaliGemma.ipynb), adjusted dataset loading and some parameters.
+
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -46,7 +37,10 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-
+TrainOutput(global_step=352, training_loss=7.797419488430023,
+metrics={'train_runtime': 1653.6164, 'train_samples_per_second': 1.705,
+'train_steps_per_second': 0.213, 'total_flos': 5.772661476596784e+16,
+'train_loss': 7.797419488430023, 'epoch': 3.9645390070921986})
 
 ### Framework versions