get_model¶
- torchvision.models.get_model(name: str, **config: Any) Module [source]¶
Gets the model name and configuration and returns an instantiated model.
Warning
The function is in Beta stage, and backward compatibility is not guaranteed.
- 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)