Model save

README.md

@@ -1,5 +1,7 @@
 ---
 library_name: transformers
+license: apache-2.0
+base_model: dandelin/vilt-b32-mlm
 tags:
 - generated_from_trainer
 model-index:
@@ -12,7 +14,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # results_outputs
 
-This model was trained from scratch on an unknown dataset.
+This model is a fine-tuned version of [dandelin/vilt-b32-mlm](https://huggingface.co/dandelin/vilt-b32-mlm) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.7022
 
 ## Model description
 
@@ -39,6 +43,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 5
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 0.5189        | 1.0   | 125  | 0.4472          |
+| 0.6232        | 2.0   | 250  | 0.4579          |
+| 0.5806        | 3.0   | 375  | 0.5368          |
+| 0.2489        | 4.0   | 500  | 0.6427          |
+| 0.1012        | 5.0   | 625  | 0.7022          |
+
+
 ### Framework versions
 
 - Transformers 4.44.2

runs/Oct15_06-24-19_463216bd54b5/events.out.tfevents.1728973621.463216bd54b5.271.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f19dcb6fffd668ad5ae67de2477b8622be3dfbb343458506b537342e8f4f1414
-size 8681
+oid sha256:a4e104dbce8d55a2fd1a3338b139ba6e0e2f7c8721c5c297f2f7eeeae534d601
+size 9306