torch.save(model.state_dict(), "spam_classifier.pth")

        __
        __