malya7 commited on
Commit
6eaaecc
·
verified ·
1 Parent(s): bbe9dbd

End of training

Browse files
README.md CHANGED
@@ -15,62 +15,23 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # finetune_colqwen2-v1.0
17
 
18
- This model is a fine-tuned version of [vidore/colqwen2-base](https://huggingface.co/vidore/colqwen2-base) on the custom dataset.
19
 
20
  ## Model description
21
 
22
- ColQwen is a model based on a novel model architecture and training strategy based on Vision Language Models (VLMs) to efficiently index documents from their visual features.
23
- It is a [Qwen2-VL-2B](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct) extension that generates [ColBERT](https://arxiv.org/abs/2004.12832)- style multi-vector representations of text and images.
24
- It was introduced in the paper [ColPali: Efficient Document Retrieval with Vision Language Models](https://arxiv.org/abs/2407.01449) and first released in [this repository](https://github.com/ManuelFay/colpali)
25
-
26
- This version is the untrained base version to guarantee deterministic projection layer initialization.
27
- <p align="center"><img width=800 src="https://github.com/illuin-tech/colpali/blob/main/assets/colpali_architecture.webp?raw=true"/></p>
28
-
29
-
30
- ## Usage
31
-
32
- ```
33
- model = ColQwen2.from_pretrained(
34
- 'toxic-pandas/finetune_colqwen2-v1.0',
35
- torch_dtype=torch.bfloat16,
36
- device_map=device,
37
- )
38
- ```
39
-
40
- ## Limitations
41
-
42
- - Focus: The model primarily focuses on PDF-type documents and high-ressources languages, potentially limiting its generalization to other document types or less represented languages.
43
- - Support: The model relies on multi-vector retreiving derived from the ColBERT late interaction mechanism, which may require engineering efforts to adapt to widely used vector retrieval frameworks that lack native multi-vector support.
44
-
45
- ## Dataset
46
-
47
- With the help of the GT4-o mini model, a dataset was formed for the completion of the colqwen2-v1.0 model, containing the following fields:
48
-
49
- - document_filename: Filename of the document.
50
- - document_url: Original URL of the document.
51
- - search_query: The query used to fetch the document.
52
- - search_topic: Topic related to the document.
53
- - search_subtopic: Subtopic related to the document.
54
- - search_language: Language specified for the search.
55
- - search_filetype: Filetype filter applied during the search.
56
- - page_number: The page's number within the document.
57
- - page_description: A natural language description of the page.
58
- - page_language: Language used on the page.
59
- - page_contains_table: Boolean indicating the presence of tables.
60
- - page_contains_figure: Boolean indicating the presence of figures.
61
- - page_contains_paragraph: Boolean indicating the presence of paragraphs.
62
- - page_image: The image of the current page.
63
- - query_type: Type of query (e.g., Extractive, Open-ended, Boolean, Compare-contrast, Enumerative, Numerical).
64
- - query_answerability: Answerability level of the query (Fully answerable, Partially answerable, Unanswerable).
65
- - query_modality: Modality used for query generation.
66
- - query_language: Language of the query.
67
- - query_reasoning: Reasoning traces used in query generation.
68
- - query: The actual query text.
69
- - query_is_self_contained: Boolean indicating if the query is self-contained.
70
- - query_is_self_contained_reasoning: Reasoning traces for determining self-contained nature.
71
- - answer: Expected answer to the question from the "query" field.
72
-
73
- ## Training hyperparameters
74
 
75
  The following hyperparameters were used during training:
76
  - learning_rate: 3e-05
@@ -84,6 +45,13 @@ The following hyperparameters were used during training:
84
  - lr_scheduler_warmup_steps: 100
85
  - num_epochs: 1
86
 
 
 
 
 
 
 
 
87
  ### Framework versions
88
 
89
  - Transformers 4.46.3
 
15
 
16
  # finetune_colqwen2-v1.0
17
 
18
+ This model is a fine-tuned version of [vidore/colqwen2-base](https://huggingface.co/vidore/colqwen2-base) on the None dataset.
19
 
20
  ## Model description
21
 
22
+ More information needed
23
+
24
+ ## Intended uses & limitations
25
+
26
+ More information needed
27
+
28
+ ## Training and evaluation data
29
+
30
+ More information needed
31
+
32
+ ## Training procedure
33
+
34
+ ### Training hyperparameters
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 3e-05
 
45
  - lr_scheduler_warmup_steps: 100
46
  - num_epochs: 1
47
 
48
+ ### Training results
49
+
50
+ | Training Loss | Epoch | Step | Validation Loss | Model Preparation Time |
51
+ |:-------------:|:------:|:----:|:---------------:|:----------------------:|
52
+ | No log | 0.0150 | 1 | 0.0268 | 0.0318 |
53
+
54
+
55
  ### Framework versions
56
 
57
  - Transformers 4.46.3
runs/Dec07_23-42-45_l-0037930f-7a7f-43a1-be61-b29649ff25d0/events.out.tfevents.1733615727.l-0037930f-7a7f-43a1-be61-b29649ff25d0.8134.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfa00a6b8ec47f42b6b2d6f6be1451bd053380cd99738e4c8b92497a2d1de41b
3
+ size 6463