Skip to content
Snippets Groups Projects
Unverified Commit 5501c7a6 authored by ZZK's avatar ZZK Committed by GitHub
Browse files

remove outdated code in conv3d (#5696)


* remove experimental

* add test num

Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent 4c7dae8f
No related branches found
No related tags found
No related merge requests found
...@@ -659,7 +659,7 @@ class Conv3d(Module): ...@@ -659,7 +659,7 @@ class Conv3d(Module):
groups=1, groups=1,
) )
) )
res = flow.experimental.cat(out_list, dim=in_channel_axis) res = flow.cat(out_list, dim=in_channel_axis)
else: else:
res = flow.F.conv3d( res = flow.F.conv3d(
x, x,
......
...@@ -21,7 +21,7 @@ from automated_test_util import * ...@@ -21,7 +21,7 @@ from automated_test_util import *
@flow.unittest.skip_unless_1n1d() @flow.unittest.skip_unless_1n1d()
class TestConv3DModule(flow.unittest.TestCase): class TestConv3DModule(flow.unittest.TestCase):
@autotest(n=2) @autotest(n=10)
def test_conv3d_with_random_data(test_case): def test_conv3d_with_random_data(test_case):
channels = random(1, 6) channels = random(1, 6)
m = torch.nn.Conv3d( m = torch.nn.Conv3d(
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment