ertk.pytorch.models.get_pt_model

ertk.pytorch.models.get_pt_model(name: str, **kwargs) → Module

Get a PyTorch model by name.

Parameters:

name: str

The model name. This should be the name of the module that implements the model.

**kwargs:

Keyword arguments to provide to the model creation function.

Returns:

model: Module

The PyTorch model.