add unit test for inverted_residual: debug 3

2020-08-11 20:34:41 +08:00 · 2020-08-11 20:34:41 +08:00 · 66224e96c5
parent db364ef26a
commit 66224e96c5
1 changed files with 0 additions and 5 deletions
--- a/tests/test_ops/test_inverted_residual_module.py
+++ b/tests/test_ops/test_inverted_residual_module.py
@ -1,6 +1,5 @@
 import pytest
 import torch
-import torch.nn as nn

 from mmseg.ops import InvertedResidual

@ -39,7 +38,3 @@ def test_inv_residual():
    x = torch.rand(1, 32, 64, 64)
    output = inv_module(x)
    assert output.shape == (1, 32, 64, 64)
-
-
-
-