Как установить переменную в качестве атрибута?

Question

Я получаю сообщение об ошибке при попытке установить переменную в качестве атрибута.

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 строке.

Answer 1

Вы хотите получить доступ к внутреннему диктату для обновления:

model = models.__dict__[args.arch](pretrained=True)

или используя getattr:

 getattr(models, args.arch)(pretrained=True)