CollaborativeCoding.load_model

Functions

load_model(→ torch.nn.Module)

Load the model based on the model name.

Module Contents

CollaborativeCoding.load_model.load_model(modelname: str, *args, **kwargs) → torch.nn.Module

Load the model based on the model name.

Args

modelnamestr

Name of the model to load.

*argslist

Additional arguments for the model class.

**kwargsdict

Additional keyword arguments for the model class.

Returns

modeltorch.nn.Module

Model object.

Raises

NotImplementedError

If the model is not implemented.

Examples

>>> from CollaborativeCoding import load_model
>>> model = load_model("magnusmodel", num_classes=10)
>>> model
MagnusModel(
  (fc1): Linear(in_features=784, out_features=100, bias=True)
  (fc2): Linear(in_features=100, out_features=10, bias=True