cglez/bert-trec-uncased

Model Card: BERT-TREC

An in-domain BERT-base model, pre-trained from scratch on the TREC dataset text.

Model Details

Description

This model is based on the BERT base (uncased) architecture and was pre-trained from scratch (in-domain) using the text in TREC dataset, excluding its test split. Only the masked language modeling (MLM) objective was used during pre-training.

• Developed by: Cesar Gonzalez-Gutierrez
• Funded by: ERC
• Architecture: BERT-base
• Language: English
• License: Apache 2.0
• Base model: BERT base model (uncased)

Checkpoints

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

Epoch	Step	Tags
1	51	epoch-1	step-51
5	256	epoch-5	step-256
10	513	epoch-10	step-513
20	1026	epoch-20	step-1026
40	2053	epoch-40	step-2053
60	3080	epoch-60	step-3080
80	4106	epoch-80	step-4106
100	5133	epoch-100	step-5133
120	6160	epoch-120	step-6160
140	7186	epoch-140	step-7186
160	8213	epoch-160	step-8213
180	9240	epoch-180	step-9240
199	10200	epoch-199	step-10200

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

from transformers import AutoModelForMaskedLM

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: 32
• Gradient accumulation steps: 3

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 Tesla V100 PCIE 32GB
• Runtime: 4 h
• Cluster Provider: Artemisa
• Compute Region: EU
• Carbon Emitted: 0.74 kg CO2 eq.

Citation

BibTeX:

[More Information Needed]