49ec0bd72b85be55579ae8ceb278c66145f593e1,mnist_hogwild/main.py,,,#,49

Before Change

    use_cuda = args.cuda and torch.cuda.is_available()
    device = torch.device("cuda" if use_cuda else "cpu")
    dataloader_kwargs = {"pin_memory": True} if use_cuda else {}

    torch.manual_seed(args.seed)
    mp.set_start_method("spawn")

After Change

    use_cuda = args.cuda and torch.cuda.is_available()
    device = torch.device("cuda" if use_cuda else "cpu")
    transform=transforms.Compose([
        transforms.ToTensor(),
        transforms.Normalize((0.1307,), (0.3081,))
        ])
    dataset1 = datasets.MNIST("../data", train=True, download=True,
                       transform=transform)
    dataset2 = datasets.MNIST("../data", train=False,
                       transform=transform)
    kwargs = {"batch_size": args.batch_size,
              "shuffle": True}
    if use_cuda:
        kwargs.update({"num_workers": 1,

Instances