def __init__(self):
    super().__init__()
    self.logit_scale = nn.Parameter((th.ones([]) * np.log((1 / 0.07))))
    self.max_log_temp = np.log(100)
def __init__(self):
    super().__init__()
    self.logit_scale = nn.Parameter((th.ones([]) * np.log((1 / 0.07))))
    self.max_log_temp = np.log(100)