7b27ce2b236e29d82e33c5e07645f845d433973c,python/ray/util/sgd/torch/torch_trainer.py,TorchTrainer,save,#TorchTrainer#Any#,520

Before Change

        Returns:
            checkpoint (str): Path to target checkpoint file.
        
        state = ray.get(self.workers[0].get_state.remote())
        torch.save(state, checkpoint)
        return checkpoint

After Change

        Returns:
            checkpoint (str): Path to target checkpoint file.
        
        torch.save(self.state_dict(), checkpoint)
        return checkpoint

    def restore(self, checkpoint):

Instances