Skip to content
Permalink
Browse files

200 epochs

  • Loading branch information...
rusty1s committed Mar 19, 2019
1 parent 488c653 commit 439f55891a09c1c7a2a176419a9fc3b6c4843d3a
Showing with 6 additions and 6 deletions.
  1. +4 −4 README.md
  2. +1 −1 benchmark/runtime/dgl/main.py
  3. +1 −1 benchmark/runtime/main.py
@@ -107,11 +107,11 @@ In addition, PyTorch Geometric is **fast**, even compared to other deep graph ne
</tr>
<tr>
<td>MUTAG</td>
<td align="center">50</td>
<td align="center">200</td>
<td align="center">RGCN</td>
<td align="right">4.64s</td>
<td align="right">0.60s</td>
<td align="right"><b>0.53s</b></td>
<td align="right">18.81s</td>
<td align="right">2.40s</td>
<td align="right"><b>2.14s</b></td>
</tr>
</tbody>
<tfoot>
@@ -49,5 +49,5 @@
g.edata.update({'type': edge_type, 'norm': edge_norm})
g.ndata['id'] = torch.arange(d.num_nodes, dtype=torch.long, device=device)
model = Net(g, d.num_nodes, d.num_classes, d.num_rels)
t = train_runtime(model, d, epochs=50, device=device)
t = train_runtime(model, d, epochs=200, device=device)
print('{} - {}: {:.2f}s'.format(d.name, Net.__name__, t))
@@ -27,5 +27,5 @@

for d, Net in product([MUTAG], [RGCN]):
model = Net(d[0].num_nodes, d.num_classes, d.num_relations)
t = train_runtime(model, d[0], epochs=50, device=device)
t = train_runtime(model, d[0], epochs=200, device=device)
print('{} - {}: {:.2f}s'.format(d.__repr__()[:-2], Net.__name__, t))

0 comments on commit 439f558

Please sign in to comment.
You can’t perform that action at this time.