fix
Browse files
app.py
CHANGED
|
@@ -30,6 +30,8 @@ names = ['LABOR-1', 'LABOR-0', 'NS']
|
|
| 30 |
|
| 31 |
indices = th.arange(g.num_nodes()).to(device)
|
| 32 |
|
|
|
|
|
|
|
| 33 |
loaders = [dgl.dataloading.DataLoader(g, indices, sampler, batch_size=batch_size, shuffle=True, drop_last=True) for sampler in samplers]
|
| 34 |
|
| 35 |
def get_time():
|
|
|
|
| 30 |
|
| 31 |
indices = th.arange(g.num_nodes()).to(device)
|
| 32 |
|
| 33 |
+
batch_size=1024
|
| 34 |
+
|
| 35 |
loaders = [dgl.dataloading.DataLoader(g, indices, sampler, batch_size=batch_size, shuffle=True, drop_last=True) for sampler in samplers]
|
| 36 |
|
| 37 |
def get_time():
|