示例#1
0
    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)
示例#2
0
文件： camel.py 项目： pcy1302/TapEM 
 def __init__(self, args):
     embedder.__init__(self, args)
示例#3
0
 def __init__(self, args):
     embedder.__init__(self, args)
     self.clip_max = torch.FloatTensor([1.0]).to(self.device)
示例#4
0
 def __init__(self, args):
     embedder.__init__(self, args)
     self.args = args
     self.coef_l = self.args.coef_layers
     self.criteria = nn.BCEWithLogitsLoss()