gpt4 book ai didi

python - 有没有办法通过字符串加载 torchvision 模型?

转载 作者:行者123 更新时间:2023-12-04 02:26:16 27 4
gpt4 key购买 nike

<分区>

目前,我使用以下代码加载预训练的 torchvision 模型:

import torchvision
torchvision.models.resnet101(pretrained=True)

但是,我喜欢将模型名称作为字符串参数,然后使用该字符串加载预训练模型。这样做的伪代码类似于:

model_name = 'resnet101'
torchvision.models.get(model_name)(pretrained=True)

有没有办法以一种相当简单的方式实现这一点?

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com