Update README.md

README.md

@@ -2,66 +2,94 @@
 library_name: transformers
 language: en
 license: mit
-datasets: []
-tags: []
+datasets:
+- community-datasets/ohsumed
+base_model:
+- openai-community/gpt2
 ---
 
-# Model Card for <Model>
+# Model Card: GPT-2-DAPT-Ohsumed
 
-A pretrained GPT2 using <Dataset>.
+A domain-adapted GPT-2, further pre-trained on the Ohsumed dataset text.
 
 ## Model Details
 
-### Model Description
+### Description
 
-A pretrained GPT2 using <Dataset>.
+This model is based on the [GPT-2](https://huggingface.co/openai-community/gpt2)
+architecture and was further pre-trained (domain-adapted) using the text in Ohsumed dataset, excluding its test split.
 
 - **Developed by:** [Cesar Gonzalez-Gutierrez](https://ceguel.es)
 - **Funded by:** [ERC](https://erc.europa.eu)
-- **Model type:** pretrained GPT2
-- **Language(s) (NLP):** English
+- **Architecture:** GPT-2
+- **Language:** English
 - **License:** MIT
-- **Pretrained from model:** [GPT2](https://huggingface.co/openai-community/gpt2)
+- **Base model:** [GPT-2](https://huggingface.co/openai-community/gpt2)
 
-### Model Checkpoints
+### Checkpoints
 
-[More Information Needed]
-
-### Model Sources
+Intermediate checkpoints from the pre-training process are available and can be accessed using specific tags,
+which correspond to training epochs and steps:
 
-- **Paper:** [More Information Needed]
+| Epoch | Step | Tags | |
+|---|---|---|---|
+| 1 | 97 | epoch-1 | step-97 |
+| 5 | 489 | epoch-5 | step-489 |
+| 10 | 978 | epoch-10 | step-978 |
+| 20 | 1956 | epoch-20 | step-1956 |
+| 40 | 3913 | epoch-40 | step-3913 |
+| 60 | 5870 | epoch-60 | step-5870 |
+| 80 | 7826 | epoch-80 | step-7826 |
+| 100 | 9783 | epoch-100 | step-9783 |
+| 120 | 11740 | epoch-120 | step-11740 |
+| 140 | 13696 | epoch-140 | step-13696 |
+| 160 | 15653 | epoch-160 | step-15653 |
+| 180 | 17610 | epoch-180 | step-17610 |
+| 198 | 19400 | epoch-198 | step-19400 |
 
-## Intended Uses & Limitations
+To load a model from a specific intermediate checkpoint, use the `revision` parameter with the corresponding tag:
+```python
+from transformers import AutoModelForCausalLM
 
-See <https://huggingface.co/openai-community/gpt2#intended-uses--limitations>.
+model = AutoModelForMaskedLM.from_pretrained("<model-name>", revision="<checkpoint-tag>")
+```
 
-### Loading Checkpoints
+### Sources
 
-[More Information Needed]
+- **Paper:** [Information pending]
 
 ## Training Details
 
-### Training Data
-
-[More Information Needed]
+For more details on the training procedure, please refer to the base model's documentation:
+[Training procedure](https://huggingface.co/openai-community/gpt2#training-procedure).
 
-#### Preprocessing [optional]
+### Training Data
 
-[More Information Needed]
+All texts from Ohsumed dataset, excluding the test partition.
 
 #### Training Hyperparameters
 
-- **Training regime:** fp16
+- **Precision:** fp16
 - **Batch size:** 8
 - **Gradient accumulation steps:** 12
 
+## Uses
+
+For typical use cases and limitations, please refer to the base model's guidance: 
+[Inteded uses & limitations](https://huggingface.co/openai-community/gpt2#intended-uses--limitations).
+
+## Bias, Risks, and Limitations
+
+This model inherits potential risks and limitations from the base model. Refer to:
+[Limitations and bias](https://huggingface.co/openai-community/gpt2#limitations-and-bias).
+
 ## Environmental Impact
 
 - **Hardware Type:** NVIDIA A100 PCIE 40GB
-- **Hours used:** [More Information Needed]
+- **Runtime:** 35 h
 - **Cluster Provider:** [Artemisa](https://artemisa.ific.uv.es/web/)
 - **Compute Region:** EU
-- **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). -->
+- **Carbon Emitted:** 5.42 kg CO2 eq.
 
 ## Citation