NTXentLoss

NTXentLoss class

dlf.losses.ntxent_loss.NTXentLoss(batch_size, temperature=0.5)

Implementation of NTXentLoss like used in SimCLR

Arguments

• batch_size: int. Used batch size
• temperature: float. Temperature to scale features

YAML Configuration

loss:
    NTXentLoss:
        batch_size: 16
        temperature: 0.5

References

• SimCLR
• Tensorflow implementation

---