def __init__(self, args): embedder.__init__(self, args) self.args = args self.coef_l = self.args.coef_layers self.coef_f = self.args.coef_fusion self.criteria = nn.BCEWithLogitsLoss() if not os.path.exists(self.args.save_root): os.makedirs(self.args.save_root)
def __init__(self, args): embedder.__init__(self, args)
def __init__(self, args): embedder.__init__(self, args) self.clip_max = torch.FloatTensor([1.0]).to(self.device)
def __init__(self, args): embedder.__init__(self, args) self.args = args self.coef_l = self.args.coef_layers self.criteria = nn.BCEWithLogitsLoss()