How to call Python function by name dynamically using a string?

Question

I have a Python function call like so:

import torchvision

model = torchvision.models.resnet18(pretrained=configs.use_trained_models)

Which works fine.

If I attempt to make it dynamic:

import torchvision

model_name = 'resnet18'
model = torchvision.models[model_name](pretrained=configs.use_trained_models)

then it fails with:

TypeError: 'module' object is not subscriptable

Which makes sense since model is a module which exports a bunch of things, including the resnet functions:

# __init__.py for the "models" module

...
from .resnet import * 
...

How can I call this function dynamically without knowing ahead of time its name (other than that I get a string with the function name)?

Jab · Accepted Answer · 2021-02-02 23:10:15Z

3

You can use the getattr function:

import torchvision

model_name = 'resnet18'
model = getattr(torchvision.models, model_name)(pretrained=configs.use_trained_models)

This essentially is along the lines the same as the dot notation just in function form accepting a string to retrieve the attribute/method.

answered Feb 2, 2021 at 23:10

Jab

27.6k22 gold badges81 silver badges117 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Edy Bourne Over a year ago

Perfect! Thanks a lot!

Shital Shah · Accepted Answer · 2023-02-14 01:58:12Z

2

The new APIs since Aug 2022 are as follows:

# returns list of available model names
torchvision.models.list_models()

# returns specified model with pretrained common weights
torchvision.models.get_model("alexnet", weights="DEFAULT")

# returns specified model with pretrained=False
torchvision.models.get_model("alexnet", weights=None)

# returns specified model with specified pretrained weights
torchvision.models.get_model("alexnet", weights=ResNet50_Weights.IMAGENET1K_V2)

Reference: https://pytorch.org/blog/easily-list-and-initialize-models-with-new-apis-in-torchvision/

answered Feb 14, 2023 at 1:58

Shital Shah

69.9k21 gold badges258 silver badges201 bronze badges

Collectives™ on Stack Overflow

How to call Python function by name dynamically using a string?

2 Answers 2

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related