@option(Configs.attention_reconstruction_loss)
def attention_reconstruction_loss(c: Configs):
    '\n    ### Initialize the attention reconstruction loss\n    '
    return AttentionReconstructionLoss(c.model.transformer.layers)
### Initialize the attention reconstruction loss
@option(Configs.attention_reconstruction_loss)
def attention_reconstruction_loss(c: Configs):
    '\n    \n    '
    return AttentionReconstructionLoss(c.model.transformer.layers)