Skip to content

GitLab

Unverified Commit 60a5b273 authored by moto's avatar moto Committed by GitHub
Browse files

Set manual seed in mask_along_axis_iid (#529)

parent d41d30ab
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
test/test_functional.py
View file @ 60a5b273
...@@ -361,13 +361,12 @@ def test_mask_along_axis(specgram, mask_param, mask_value, axis): ...@@ -361,13 +361,12 @@ def test_mask_along_axis(specgram, mask_param, mask_value, axis):
assert num_masked_columns < mask_param assert num_masked_columns < mask_param
@pytest.mark.parametrize('specgrams', [
torch.randn(4, 2, 1025, 400),
])
@pytest.mark.parametrize('mask_param', [100]) @pytest.mark.parametrize('mask_param', [100])
@pytest.mark.parametrize('mask_value', [0., 30.]) @pytest.mark.parametrize('mask_value', [0., 30.])
@pytest.mark.parametrize('axis', [2, 3]) @pytest.mark.parametrize('axis', [2, 3])
def test_mask_along_axis_iid(specgrams, mask_param, mask_value, axis): def test_mask_along_axis_iid(mask_param, mask_value, axis):
torch.random.manual_seed(42)
specgrams = torch.randn(4, 2, 1025, 400)
mask_specgrams = F.mask_along_axis_iid(specgrams, mask_param, mask_value, axis) mask_specgrams = F.mask_along_axis_iid(specgrams, mask_param, mask_value, axis)
......
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