athena.models.kws.transformer_wakeup_focal_loss

speech transformer implementation

Module Contents

Classes

KWSTransformer_FocalLoss

Standard implementation of a KWSTransformer. Model mainly consists of three parts:

class athena.models.kws.transformer_wakeup_focal_loss.KWSTransformer_FocalLoss(data_descriptions, config=None)

Bases: athena.models.kws.base.BaseModel

Standard implementation of a KWSTransformer. Model mainly consists of three parts: the x_net for input preparation, the transformer itself

default_config

call(samples, training=None)

call model

build_model(data_descriptions)

get_loss(outputs, samples, training=None)

get loss