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

Question

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

from transformers import BertModel, BertTokenizer, BertConfig

config=BertConfig.from_pretrained(arch)

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

Answer 1

是的，你可以

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