48cdca843f2d06aeb356db27d48e01c1153eba22,python/ray/util/sgd/pytorch/examples/dcgan.py,,train_example,#Any#Any#Any#,214

Before Change

def train_example(num_replicas=1, use_gpu=False, test_mode=False):
    config = {TEST_MODE: test_mode}
    trainer = PyTorchTrainer(
        model_creator,
        data_creator,
        optimizer_creator,

After Change

def train_example(num_replicas=1, use_gpu=False, test_mode=False):
    config = {
        "test_mode": test_mode,
        "classification_model_path": os.path.join(
            os.path.dirname(ray.__file__),
            "util/sgd/pytorch/examples/mnist_cnn.pt")
    }
    trainer = PyTorchTrainer(
        model_creator,
        data_creator,
        optimizer_creator,

Instances