0ad33d606682537466f3430fc6d6ac7d47460f1a,beginner_source/blitz/data_parallel_tutorial.py,,,#,118

Before Change

// 

for data in rand_loader:
    if torch.cuda.is_available():
        input_var = Variable(data.cuda())
    else:
        input_var = Variable(data)

    output = model(input_var)
    print("Outside: input size", input_var.size(),
          "output_size", output.size())

After Change

//

for data in rand_loader:
    input = data.to(device)
    output = model(input)
    print("Outside: input size", input.size(),
          "output_size", output.size())

Instances