sakura.model_controllers.extractor_controller.ExtractorController.select_loss_dict

ExtractorController.select_loss_dict(selection=None, internal=None)

Select loss keys from the internal dict based on selection setting.

Parameters:

  • selection (list or '*') – The selection setting, which can be a list of keys, ‘*’ for all keys

  • internal (dict) – The internal loss configuration dictionary to select from.

Returns:

A list of selected keys based on the selection criteria.

Return type:

list