def _save_model()

in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def _save_model(self):
        torch.save(
            self.model.state_dict(), os.path.join(self.output_model_path, "model.pt")
        )