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