Method save_bert

bing_bert/turing/models.py:133–134

(self, filename: str)

        self.network.train()

    def save_bert(self, filename: str):
        return torch.save(self.bert_encoder.state_dict(), filename)

    def to(self, device):
        assert isinstance(device, torch.device)