Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "1cc3fbe975634cd39141be2e460a09388cae3f41"
Commit 0003e008 authored by rusty1s's avatar rusty1s
Browse files

walk length increased

parent 1dcb3933
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_rw.py
View file @ 0003e008
...@@ -11,7 +11,7 @@ def test_rw(): ...@@ -11,7 +11,7 @@ def test_rw():
start = tensor([0, 1, 2, 3, 4], torch.long, device) start = tensor([0, 1, 2, 3, 4], torch.long, device)
row = tensor([0, 1, 1, 1, 2, 2, 3, 3, 4, 4], torch.long, device) row = tensor([0, 1, 1, 1, 2, 2, 3, 3, 4, 4], torch.long, device)
col = tensor([1, 0, 2, 3, 1, 4, 1, 4, 2, 3], torch.long, device) col = tensor([1, 0, 2, 3, 1, 4, 1, 4, 2, 3], torch.long, device)
walk_length = 2 walk_length = 10
out = random_walk(row, col, start, walk_length) out = random_walk(row, col, start, walk_length)
assert out[:, 0].tolist() == start.tolist() assert out[:, 0].tolist() == start.tolist()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment