Skip to content

Commit eec52d4

Browse files
update code
1 parent ecdc38e commit eec52d4

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

data.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,3 @@ def get_iterators(args, device):
4242
sort=False, sort_key=lambda x: len(x.nodes), sort_within_batch=False, repeat=False, shuffle=False)
4343
print("Done")
4444
return train_iter, test_iter
45-
46-
opt = config.parse()
47-
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
48-
train_iter, test_iter = get_iterators(opt, device)

0 commit comments

Comments
 (0)