57e63487536fe9493091c4d1b011a8760c805df4,examples/shapenet.py,,train,#Any#,96

Before Change

    model.train()

    for data in train_loader:
        data = data.cuda(["input", "pos", "index", "weight", "target"])
        data = data.to_variable()
        optimizer.zero_grad()
        output = model(data.input, data.pos, data.adj)
        print(output.size())
        loss = F.nll_loss(output, data.target)
        loss.backward()

After Change

    for data in train_loader:
        data = data.cuda().to_variable()
        optimizer.zero_grad()
        output = model(data)
        loss = F.nll_loss(output, data.target)
        loss.backward()
        optimizer.step()

Instances