Add nn modules block tests

1 year ago · 8ede2d1e9c
parent 8a970f23d3
commit 8ede2d1e9c
1 changed files with 1 additions and 1 deletions
--- a/tests/test_python.py
+++ b/tests/test_python.py
@ -387,7 +387,7 @@ def test_nn_modules_conv():
    DWConvTranspose2d(c1, c2)(x)
    ConvTranspose(c1, c2)(x)
    Focus(c1, c2)(x)
-    CBAM(c1, c2)(x)
+    CBAM(c1)(x)

    # Fuse ops
    m = Conv2(c1, c2)