attribench.BasicModelFactory

class attribench.BasicModelFactory(base_model)[source]

Bases: ModelFactory

Basic implementation of a ModelFactory that just returns a deep copy of a given model. This can be used for simple models, but requires the model to be picklable (e.g. lambda layers won’t work and require a specific implementation of ModelFactory)

Parameters:

base_modelnn.Module: Model of which to return a deep copy.

Methods

__call__()[source]

Return a deep copy of the base model.

Returns:

nn.Module: Deep copy of the base model.

Return type:

Module