mirrors/mmpretrain
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmpretrain.git synced 2025-06-03 14:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix linting

Browse Source
This commit is contained in:
lixiaojie 2020-06-14 01:25:08 +08:00
parent fb3934fd2c
commit 85844a3a9e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_backbones/test_shufflenet_v1.py
View File

@ -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()