【发布时间】:2020-11-21 19:14:57
【问题描述】:
我正在尝试使用 Transformer 的拥抱脸预训练模型 bert-base-uncased,但我想增加 dropout。 from_pretrained 方法中没有提到这一点,但是 colab 运行下面的对象实例化没有任何问题。我在 classtransformers.BertConfig 文档中看到了这些 dropout 参数。
我是否以正确的方式使用了 bert-base-uncased AND 更改 dropout?
model = BertForSequenceClassification.from_pretrained(
pretrained_model_name_or_path='bert-base-uncased',
num_labels=2,
output_attentions = False,
output_hidden_states = False,
attention_probs_dropout_prob=0.5,
hidden_dropout_prob=0.5
)
【问题讨论】:
标签: python bert-language-model huggingface-transformers