415be78cc0d1275a29d0ceda550d0d7a7a5224ea,python/ray/util/sgd/tests/test_torch.py,,test_save_and_restore,#Any#Any#Any#,560

Before Change

    trainer1.shutdown()

    trainer2 = TorchTrainer(
        model_creator=model_creator,
        data_creator=data_creator,
        optimizer_creator=optimizer_creator,
        loss_creator=lambda config: nn.MSELoss(),
        num_workers=num_workers)
    trainer2.load(checkpoint_path)

    model2 = trainer2.get_model()

After Change

    trainer1.shutdown()

    trainer2 = TorchTrainer(
        training_operator_cls=Operator, num_workers=num_workers)
    trainer2.load(checkpoint_path)

    model2 = trainer2.get_model()

Instances