BERT L-2 H-512 fine-tuned on MLM (CORD-19 2020/06/16)

BERT model with 2 Transformer layers and hidden embedding of size 512, referenced in Well-Read Students Learn Better: On the Importance of Pre-training Compact Models, fine-tuned for MLM on CORD-19 dataset (as released on 2020/06/16).

Training the model

    --model_type bert
    --model_name_or_path google/bert_uncased_L-2_H-512_A-8
    --train_data_file {cord19-200616-dataset}
    --mlm_probability 0.2
    --block_size 512
    --per_device_train_batch_size 20
    --learning_rate 3e-5
    --num_train_epochs 2
    --output_dir bert_uncased_L-2_H-512_A-8_cord19-200616

Select AutoNLP in the “Train” menu to fine-tune this model automatically.

Downloads last month
Hosted inference API
Mask token: [MASK]
This model can be loaded on the Inference API on-demand.