From 03259ee453eefddfc1a7e0339f99779e05fa2d76 Mon Sep 17 00:00:00 2001 From: WangYi Date: Thu, 14 Dec 2023 04:33:45 +0000 Subject: [PATCH] recover modification --- python/oneflow/test/expensive/test_compatibility.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/oneflow/test/expensive/test_compatibility.py b/python/oneflow/test/expensive/test_compatibility.py index 4d25b153689..0e1d6b23733 100644 --- a/python/oneflow/test/expensive/test_compatibility.py +++ b/python/oneflow/test/expensive/test_compatibility.py @@ -29,10 +29,10 @@ def test_resnet50_compatibility(test_case): test_case, "pytorch_resnet.py", "resnet50", "cuda", 16, 224 ) - # def test_convmixer_compatibility(test_case): - # do_test_train_loss_oneflow_pytorch( - # test_case, "pytorch_convmixer.py", "convmixer_768_32_relu", "cuda", 4, 224 - # ) + def test_convmixer_compatibility(test_case): + do_test_train_loss_oneflow_pytorch( + test_case, "pytorch_convmixer.py", "convmixer_768_32_relu", "cuda", 4, 224 + ) def test_densenet_compatibility(test_case): do_test_train_loss_oneflow_pytorch(