尝试将变量设置为属性时出现错误。
parser = argparse.ArgumentParser()
parser.add_argument('--arch', action='store',
dest='arch', default='alexnet',
help='Store a simple value')
args = parser.parse_args()
model = models.args.arch(pretrained=True)
我知道 models.args.arch
会产生错误,但是将变量设置为属性的语法应该是什么样的?我可以用 if 语句来做到这一点,但这将是很多代码,我想这可能在 1 行中。
最佳答案
您想要访问内部字典来更新:
model = models.__dict__[args.arch](pretrained=True)
或使用getattr
:
getattr(models, args.arch)(pretrained=True)
关于python - 如何将变量设置为属性?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54012820/