时间戳

首页 > 解决方案 > 我可以在 Python 中的导入语句和函数调用中使用变量吗？

python - 我可以在 Python 中的导入语句和函数调用中使用变量吗？

问题描述

我正在创建用于训练神经网络模型的代码，并希望使其具有通用性，以便我可以更改模型名称并为每个模型运行代码。我可以在 Python 中的导入语句和函数调用中使用变量吗？

from transformers import BertModel, BertTokenizer, BertConfig

config=BertConfig.from_pretrained(arch)

在这些行中，我想要一个变量名而不是我可以更改的 Bert（Bert <--> CTRL <--> 任何其他模型）

标签： pythonneural-network

解决方案

是的你可以

from transformers import BertConfig as btcfg
config=btcfg.from_pretrained(arch)

推荐阅读