cglez/gpt2-dapt-trec

Model Card: GPT-2-DAPT-TREC

A domain-adapted GPT-2, further pre-trained on the TREC dataset text.

Model Details

Description

This model is based on the GPT-2 architecture and was further pre-trained (domain-adapted) using the text in TREC dataset, excluding its test split.

• —Developed by: Cesar Gonzalez-Gutierrez
• —Funded by: ERC
• —Architecture: GPT-2
• —Language: English
• —License: MIT
• —Base model: GPT-2

Checkpoints

Intermediate checkpoints from the pre-training process are available and can be accessed using specific tags, which correspond to training epochs and steps:

To load a model from a specific intermediate checkpoint, use the revision parameter with the corresponding tag:

from transformers import AutoModelForCausalLM

model = AutoModelForMaskedLM.from_pretrained("<model-name>", revision="<checkpoint-tag>")

Sources

• —Paper: [Information pending]

Training Details

For more details on the training procedure, please refer to the base model's documentation: Training procedure.

Training Data

All texts from TREC dataset, excluding the test partition.

Training Hyperparameters

• —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.

Bias, Risks, and Limitations

This model inherits potential risks and limitations from the base model. Refer to: Limitations and bias.

Environmental Impact

• —Hardware Type: NVIDIA A100 PCIE 40GB
• —Runtime: 7 h
• —Cluster Provider: Artemisa
• —Compute Region: EU
• —Carbon Emitted: 1.08 kg CO2 eq.

Citation

BibTeX:

[More Information Needed]