3bba9a4f2dc3cecc76b105f9601a49e543f249e1,examples/data_parallel.py,,,#,29
Before Change
for data_list in loader:
output = model(data_list)
print("Outside Model: num graphs: {}".format(output.size(0)) )
After Change
for data_list in loader:
optimizer.zero_grad()
output = model(data_list)
y = torch .cat([data.y for data in data_list]).to(output.device)
loss = F.nll_loss(output, y)
loss.backward()
optimizer.step()
In pattern: SUPERPATTERN
Frequency: 3
Non-data size: 3
Instances Project Name: rusty1s/pytorch_geometric
Commit Name: 3bba9a4f2dc3cecc76b105f9601a49e543f249e1
Time: 2019-03-17
Author: matthias.fey@tu-dortmund.de
File Name: examples/data_parallel.py
Class Name:
Method Name:
Project Name: rlworkgroup/garage
Commit Name: e09e6dc678f9b9c198ed361eea0dffa7749bc553
Time: 2020-05-01
Author: 43084978+yonghyuc@users.noreply.github.com
File Name: src/garage/torch/algos/vpg.py
Class Name: VPG
Method Name: train_once
Project Name: rusty1s/pytorch_geometric
Commit Name: cce8409beb4c1b662de2b9635cd723993184be23
Time: 2017-11-10
Author: matthias.fey@tu-dortmund.de
File Name: torch_geometric/datasets/utils/faust_monet.py
Class Name:
Method Name: