get_model¶
- torchvision.models.get_model(name: str, **config: Any) Module [source]¶
Gets the model name and configuration and returns an instantiated model.
- Parameters:
name (str) – The name under which the model is registered.
**config (Any) – parameters passed to the model builder method.
- Returns:
The initialized model.
- Return type:
model (nn.Module)
Examples using
get_model
:Transforms v2: End-to-end object detection example
Transforms v2: End-to-end object detection example