src.model

Subpackages

• src.model.extras
  • Submodules
  • src.model.extras.example_module
  • Module contents

Module contents

class src.model.SeparationModel(config, extra_modules=None)

Bases: torch.nn.modules.module.Module

forward(data)

Parameters

data – (dict) a dictionary containing the input data for the model. Should match the input_keys in self.input.

Returns:

project_data(data, clamp=False)

save(location, metadata=None)

Saves a SeparationModel into a location into a dictionary with the weights and model configuration. :param location: (str) Where you want the model saved, as a path.

Returns: