b16cb3485a7a653435d08734ce752b561aba6432,pyannote/audio/embedding/approaches/classification.py,Classification,load_more,#Classification#Any#,104

Before Change

            classifier_pt = self.CLASSIFIER_PT.format(
                train_dir=self.train_dir_, epoch=self.epoch_)
        else:
            msg = "TODO: infer classifier_pt from model_pt"
            raise NotImplementedError(msg)

        classifier_state = torch.load(
            classifier_pt, map_location=lambda storage, loc: storage)
        self.classifier_.load_state_dict(classifier_state)

After Change

            classifier_pt = self.CLASSIFIER_PT.format(
                train_dir=self.train_dir_, epoch=self.epoch_)
        else:
            classifier_pt = model_pt.with_suffix(".classifier.pt")

        classifier_state = torch.load(
            classifier_pt, map_location=lambda storage, loc: storage)
        self.classifier_.load_state_dict(classifier_state)

Instances