1fd1fbb4b677f10d8e2181e7176daabad7327a72,python/baseline/pytorch/tagger/model.py,TaggerModelBase,create,#Any#Any#Any#,62

Before Change

                pytorch_linear(hsz, len(model.labels)),
            ))

        if model.use_crf:
            if model.crf_mask:
                assert model.span_type is not None, "A crf mask cannot be used without providing `span_type`"
                model.crf = CRF(
                    len(labels),
                    (Offsets.GO, Offsets.EOS), batch_first=False,
                    vocab=model.labels, span_type=model.span_type, pad_idx=Offsets.PAD
                )
            else:
                model.crf = CRF(len(labels), (Offsets.GO, Offsets.EOS), batch_first=False)
        model.crit = SequenceCriterion(LossFn=nn.CrossEntropyLoss)
        print(model)
        return model

After Change

                mask=constraint
            )
        else:
            if constraint is not None:
                constraint = F.log_softmax(torch.zeros(constraint.shape).masked_fill(constraint, -1e4), dim=0)
                model.register_buffer("constraint", constraint.unsqueeze(0))
            else:
                model.constraint = None
            model.crit = SequenceCriterion(LossFn=nn.CrossEntropyLoss)
        print(model)
        return model

Instances