def __init__()

in trainer/loss.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, use_cuda: bool, margin: float = 0.1):
        super(ContrastiveLoss_coot, self).__init__()
        self.margin = margin
        self.sim = cosine_sim
        self.use_cuda = use_cuda