1 from_pretrained
- 从预训练模型配置中实例化一个 PyTorch 预训练模型
- 默认情况下,模型使用
model.eval()
设置为评估模式(Dropout 模块被禁用)- 要训练模型,应该首先使用
model.train()
将其设置回训练模式
- 要训练模型,应该首先使用
1.1 主要参数
pretrained_model_name_or_path | 需要加载的模型,可以是:
|
from_tf | (bool, 可选,默认为 False) - 从 TensorFlow 检查点保存文件中加载模型权重 |
force_download | (bool, 可选,默认为 False) - 是否强制(重新)下载模型权重和配置文件,覆盖已存在的缓存版本 |
local_files_only | (bool, 可选,默认为 False) - 是否只查看本地文件(即,不尝试下载模型) |