fix linting

2020-06-14 01:25:08 +08:00 · 2020-06-14 01:25:08 +08:00 · 85844a3a9e
parent fb3934fd2c
commit 85844a3a9e
1 changed files with 2 additions and 2 deletions
--- a/tests/test_backbones/test_shufflenet_v1.py
+++ b/tests/test_backbones/test_shufflenet_v1.py
@ -228,7 +228,7 @@ def test_shufflenetv1_backbone():
    assert feat[1].shape == torch.Size((1, 960, 7, 7))

    # Test ShuffleNetv1 forward with layers 2 forward
-    model = ShuffleNetv1(groups=3, out_indices=(2,))
+    model = ShuffleNetv1(groups=3, out_indices=(2, ))
    model.init_weights()
    model.train()