From 5501c7a6869f04686788b4fe8ce6e70134ffba0c Mon Sep 17 00:00:00 2001
From: ZZK <42901638+MARD1NO@users.noreply.github.com>
Date: Tue, 3 Aug 2021 02:29:17 +0800
Subject: [PATCH] remove outdated code in conv3d (#5696)

* remove experimental

* add test num

Co-authored-by: oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
---
 python/oneflow/nn/modules/conv.py          | 2 +-
 python/oneflow/test/modules/test_conv3d.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/oneflow/nn/modules/conv.py b/python/oneflow/nn/modules/conv.py
index 0f4f0a9ed..70ecfe7e9 100644
--- a/python/oneflow/nn/modules/conv.py
+++ b/python/oneflow/nn/modules/conv.py
@@ -659,7 +659,7 @@ class Conv3d(Module):
                         groups=1,
                     )
                 )
-            res = flow.experimental.cat(out_list, dim=in_channel_axis)
+            res = flow.cat(out_list, dim=in_channel_axis)
         else:
             res = flow.F.conv3d(
                 x,
diff --git a/python/oneflow/test/modules/test_conv3d.py b/python/oneflow/test/modules/test_conv3d.py
index 9d2414cde..1087ab792 100644
--- a/python/oneflow/test/modules/test_conv3d.py
+++ b/python/oneflow/test/modules/test_conv3d.py
@@ -21,7 +21,7 @@ from automated_test_util import *
 
 @flow.unittest.skip_unless_1n1d()
 class TestConv3DModule(flow.unittest.TestCase):
-    @autotest(n=2)
+    @autotest(n=10)
     def test_conv3d_with_random_data(test_case):
         channels = random(1, 6)
         m = torch.nn.Conv3d(
-- 
GitLab