8f8ed00abdc2cd8eccf8308e3fc7aed833c5ae23,finetune/target_models/sequence_labeling.py,SequenceLabeler,__init__,#SequenceLabeler#,171

Before Change

        d.update(kwargs)
        super().__init__(**d)
        if "n_epochs" not in kwargs.keys():
            self.config.n_epochs = math.ceil(self.config.n_epochs * 1.5)

    def _get_input_pipeline(self):
        return SequencePipeline(
            config=self.config, multi_label=self.config.multi_label_sequences

After Change

        d.update(kwargs)
        super().__init__(**d)
        // If our basemodel has a default n_epochs > 5, use that; otherwise, use 5 epochs
        base_model_epochs = self.config.base_model.settings.get("n_epochs", 5)
        self.config.n_epochs = max(base_model_epochs, 5)

    def _get_input_pipeline(self):
        return SequencePipeline(
            config=self.config, multi_label=self.config.multi_label_sequences

Instances