config_bert: BERT Model Parameters

In macmillancontentscience/torchtransformers: Transformer Models in Torch

BERT Model Parameters

Description

Several parameters define a BERT model. This function can be used to easily load them.

Usage

config_bert(
  bert_type,
  parameter = c("embedding_size", "n_layer", "n_head", "max_tokens", "vocab_size",
    "tokenizer_scheme")
)

Arguments

bert_type	Character scalar; the name of a known BERT model.
parameter	Character scalar; the desired parameter.

Value

Integer scalar; the value of that parameter for that model.

Examples

config_bert("bert_medium_uncased", "n_head")

macmillancontentscience/torchtransformers documentation built on Aug. 6, 2023, 5:35 a.m.