diff --git a/python/oneflow/nn/modules/conv.py b/python/oneflow/nn/modules/conv.py index 0f4f0a9ed25590baa4c9d2359015e78a7732b7c1..70ecfe7e909ffaf7316f53c3547fef74640ae32b 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 9d2414cde7727971726b9d2cd8b109452d1cba1b..1087ab792204a4295dadb5664fb19d37082811c1 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(