def save_model(self, export_model):
        """Save a Semi-Supervised Deep Generative model to export_model."""

        net_dict = self.net.state_dict()
        torch.save({'net_dict': net_dict}, export_model)
Save a Semi-Supervised Deep Generative model to export_model.
adbench/baseline/DeepSAD/src/baselines/SemiDGM.py