Skip to content

Commit

Permalink
recover modification
Browse files Browse the repository at this point in the history
  • Loading branch information
marigoold committed Dec 14, 2023
1 parent 139b2cf commit 03259ee
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/oneflow/test/expensive/test_compatibility.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 03259ee

Please sign in to comment.