b696ad3275b2b7fb00fcd6b95d11d85f6d813101,nrekit/framework/bag_re.py,BagRE,eval_model,#BagRE#Any#,140

Before Change

            t = tqdm(eval_loader)
            pred_result = []
            for iter, data in enumerate(t):
                label, bag_name, scope, token, pos1, pos2, mask = data
                if torch.cuda.is_available():
                    token = token.cuda()
                    pos1 = pos1.cuda()
                    pos2 = pos2.cuda()
                    label = label.cuda()
                    mask = mask.cuda()
                logits = self.forward(None, scope, token, pos1, pos2, mask, train=False) // results after softmax
                for i in range(logits.size(0)):
                    for relid in range(self.num_class):
                        if self.id2rel[relid] != "NA":

After Change

            pred_result = []
            for iter, data in enumerate(t):
                if torch.cuda.is_available():
                    for i in range(len(data)):
                        try:
                            data[i] = data[i].cuda()
                        except:
                            pass
                label = data[0]
                bag_name = data[1]
                scope = data[2]
                args = data[3:]
                logits = self.model(None, scope, *args, train=False) // results after softmax
                for i in range(logits.size(0)):
                    for relid in range(self.model.num_class):
                        if self.model.id2rel[relid] != "NA":

Instances