renate.utils.avalanche module

class renate.utils.avalanche.BaseAvalancheDataset(targets, collate_fn=None)

Bases: Dataset

Base class for all datasets consumable by Avalanche updaters.

class renate.utils.avalanche.AvalancheDataset(dataset, targets, collate_fn=None)

Bases: BaseAvalancheDataset

A wrapper around a Dataset consumable by Avalanche updaters.

class renate.utils.avalanche.AvalancheDatasetForBuffer(buffer, targets, collate_fn=None)

Bases: BaseAvalancheDataset

A wrapper around a DataBuffer consumable by Avalanche updaters.

renate.utils.avalanche.to_avalanche_dataset(dataset, collate_fn=None)

Converts a DataBuffer or Dataset into an Avalanche-compatible Dataset.

Return type:

BaseAvalancheDataset

class renate.utils.avalanche.AvalancheBenchmarkWrapper(train_dataset, val_dataset, train_transform, train_target_transform, test_transform, test_target_transform)

Bases: object

update_benchmark_properties()

state_dict()

Returns the state of the benchmark.

Return type:

Dict[str, Any]

load_state_dict(state_dict)

Restores the state of the benchmark.

Return type:

None

renate.utils.avalanche.replace_plugin(plugin, plugins)

Replaces a plugin if already exists and appends otherwise.

Parameters:

• plugin (Optional[BasePlugin]) – New plugin that replaces existing one.

• plugins (List[BasePlugin]) – List of current plugins.

Return type:

List[BasePlugin]

Returns:

Reference to plugins.

renate.utils.avalanche.remove_plugin(plugin_class, plugins)

Removes a plugin by class if exists.

Parameters:

• plugin_class (Type[BasePlugin]) – Remove a plugin of this class if exists.

• plugins (List[BasePlugin]) – List of current plugins.

Return type:

List[BasePlugin]

Returns:

Reference to plugins.

renate.utils.avalanche.plugin_by_class(plugin_class, plugins)

Returns plugin with respective class from a list of plugins.

Parameters:

• plugin_class (Type[BasePlugin]) – Class type of interest in plugins.

• plugins (List[BasePlugin]) – List of plugins we search for an object of type plugin_class.

Return type:

Optional[BasePlugin]

Returns:

None if class does not exist, otherwise the respective object.