fix linting

2020-06-11 16:40:26 +08:00 · 2020-06-11 16:40:26 +08:00 · 4d5c39d1ee
parent 93612db881
commit 4d5c39d1ee
1 changed files with 2 additions and 3 deletions
--- a/tests/test_backbone.py
+++ b/tests/test_backbone.py
@ -118,9 +118,8 @@ def test_shufflenetv1_backbone():
    assert feat[3].shape == torch.Size([1, 960, 7, 7])

    # Test ShuffleNetv1 forward with GroupNorm forward
-    model = ShuffleNetv1(groups=3,
-                         norm_cfg=dict(type='GN', num_groups=2,
-                                       requires_grad=True))
+    model = ShuffleNetv1(
+        groups=3, norm_cfg=dict(type='GN', num_groups=2, requires_grad=True))
    model.init_weights()
    model.train()