mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add unit test for inverted_residual: debug 3

pull/58/head
johnzja 2020-08-11 20:34:41 +08:00
parent db364ef26a
commit 66224e96c5
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_ops/test_inverted_residual_module.py
View File

@ -1,6 +1,5 @@
import pytest import pytest
import torch import torch
import torch.nn as nn
from mmseg.ops import InvertedResidual from mmseg.ops import InvertedResidual
@ -39,7 +38,3 @@ def test_inv_residual():
x = torch.rand(1, 32, 64, 64) x = torch.rand(1, 32, 64, 64)
output = inv_module(x) output = inv_module(x)
assert output.shape == (1, 32, 64, 64) assert output.shape == (1, 32, 64, 64)