cglez commited on
Commit
560ce4f
·
verified ·
1 Parent(s): 8cc5d86

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -30
README.md CHANGED
@@ -2,72 +2,94 @@
2
  library_name: transformers
3
  language: en
4
  license: apache-2.0
5
- datasets: []
6
- tags: []
 
 
7
  ---
8
 
9
- # Model Card for <Model>
10
 
11
- A pretrained BERT using <Dataset>.
12
 
13
  ## Model Details
14
 
15
- ### Model Description
16
 
17
- A MLM-only pretrained BERT-base using <Dataset>.
 
 
18
 
19
  - **Developed by:** [Cesar Gonzalez-Gutierrez](https://ceguel.es)
20
  - **Funded by:** [ERC](https://erc.europa.eu)
21
- - **Model type:** MLM pretrained BERT
22
- - **Language(s) (NLP):** English
23
- - **License:** Apache license 2.0
24
- - **Pretrained from model:** [BERT base model (uncased)](https://huggingface.co/google-bert/bert-base-uncased)
25
 
26
- ### Model Checkpoints
27
 
28
- [More Information Needed]
29
-
30
- ### Model Sources
31
-
32
- - **Paper:** [More Information Needed]
33
-
34
- ## Uses
35
 
36
- See <https://huggingface.co/google-bert/bert-base-uncased#intended-uses--limitations>.
 
 
 
 
 
 
 
 
 
37
 
38
- ### Checkpoint Use
 
 
39
 
40
- [More Information Needed]
 
41
 
42
- ## Bias, Risks, and Limitations
43
 
44
- See <https://huggingface.co/google-bert/bert-base-uncased#limitations-and-bias>.
45
 
46
  ## Training Details
47
 
48
- See <https://huggingface.co/google-bert/bert-base-uncased#training-procedure>.
 
49
 
50
  ### Training Data
51
 
52
- [More Information Needed]
53
 
54
- #### Preprocessing [optional]
55
 
56
- [More Information Needed]
57
 
58
  #### Training Hyperparameters
59
 
60
- - **Training regime:** fp16
61
  - **Batch size:** 32
62
  - **Gradient accumulation steps:** 3
63
 
 
 
 
 
 
 
 
 
 
 
64
  ## Environmental Impact
65
 
66
  - **Hardware Type:** NVIDIA Tesla V100 PCIE 32GB
67
- - **Hours used:** [More Information Needed]
68
  - **Cluster Provider:** [Artemisa](https://artemisa.ific.uv.es/web/)
69
  - **Compute Region:** EU
70
- - **Carbon Emitted:** [More Information Needed] <!-- Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). -->
71
 
72
  ## Citation
73
 
 
2
  library_name: transformers
3
  language: en
4
  license: apache-2.0
5
+ datasets:
6
+ - cglez/civil_comments_clean
7
+ base_model:
8
+ - google-bert/bert-base-uncased
9
  ---
10
 
11
+ # Model Card: BERT Civil Comments
12
 
13
+ An in-domain BERT-base model, pre-trained from scratch on the Civil Comments dataset text.
14
 
15
  ## Model Details
16
 
17
+ ### Description
18
 
19
+ This model is based on the [BERT base (uncased)](https://huggingface.co/google-bert/bert-base-uncased)
20
+ architecture and was pre-trained from scratch (in-domain) using the text in Civil Comments dataset, excluding its test split.
21
+ Only the masked language modeling (MLM) objective was used during pre-training.
22
 
23
  - **Developed by:** [Cesar Gonzalez-Gutierrez](https://ceguel.es)
24
  - **Funded by:** [ERC](https://erc.europa.eu)
25
+ - **Architecture:** BERT-base
26
+ - **Language:** English
27
+ - **License:** Apache 2.0
28
+ - **Base model:** [BERT base model (uncased)](https://huggingface.co/google-bert/bert-base-uncased)
29
 
30
+ ### Checkpoints
31
 
32
+ Intermediate checkpoints from the pre-training process are available and can be accessed using specific tags,
33
+ which correspond to training epochs and steps:
 
 
 
 
 
34
 
35
+ | Epoch | Step | Tags | |
36
+ |---|---|---|---|
37
+ | 1 | 17833 | epoch-1 | step-17833 |
38
+ | 2 | 35666 | epoch-2 | step-35666 |
39
+ | 3 | 53499 | epoch-3 | step-53499 |
40
+ | 5 | 89166 | epoch-5 | step-89166 |
41
+ | 10 | 178330 | epoch-10 | step-178330 |
42
+ | 15 | 267495 | epoch-15 | step-267495 |
43
+ | 20 | 356660 | epoch-20 | step-356660 |
44
+ | 25 | 445825 | epoch-25 | step-445825 |
45
 
46
+ To load a model from a specific intermediate checkpoint, use the `revision` parameter with the corresponding tag:
47
+ ```python
48
+ from transformers import AutoModelForMaskedLM
49
 
50
+ model = AutoModelForMaskedLM.from_pretrained("<model-name>", revision="<checkpoint-tag>")
51
+ ```
52
 
53
+ ### Sources
54
 
55
+ - **Paper:** [Information pending]
56
 
57
  ## Training Details
58
 
59
+ For more details on the training procedure, please refer to the base model's documentation:
60
+ [Training procedure](https://huggingface.co/google-bert/bert-base-uncased#training-procedure).
61
 
62
  ### Training Data
63
 
64
+ All texts from Civil Comments dataset, excluding the test partition.
65
 
66
+ #### Preprocessing
67
 
68
+ All markup and symbols were removed from the texts, including punctuation.
69
 
70
  #### Training Hyperparameters
71
 
72
+ - **Precision:** fp16
73
  - **Batch size:** 32
74
  - **Gradient accumulation steps:** 3
75
 
76
+ ## Uses
77
+
78
+ For typical use cases and limitations, please refer to the base model's guidance:
79
+ [Inteded uses & limitations](https://huggingface.co/google-bert/bert-base-uncased#intended-uses--limitations).
80
+
81
+ ## Bias, Risks, and Limitations
82
+
83
+ This model inherits potential risks and limitations from the base model. Refer to:
84
+ [Limitations and bias](https://huggingface.co/google-bert/bert-base-uncased#limitations-and-bias).
85
+
86
  ## Environmental Impact
87
 
88
  - **Hardware Type:** NVIDIA Tesla V100 PCIE 32GB
89
+ - **Hours used:** 37
90
  - **Cluster Provider:** [Artemisa](https://artemisa.ific.uv.es/web/)
91
  - **Compute Region:** EU
92
+ - **Carbon Emitted:** 6.88 kg CO2 eq.
93
 
94
  ## Citation
95