github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 10:34:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx] added timm model tracing testing (#1221)

Browse Source
This commit is contained in:
Frank Lee
2022-07-07 14:02:17 +08:00
committed by GitHub
parent 280a81243d
commit b6cb5a47ad
4 changed files with 125 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_fx/test_coloproxy.py
View File

@@ -3,7 +3,6 @@ from colossalai.fx.proxy import ColoProxy
import pytest
@pytest.mark.skip
def test_coloproxy():
# create a dummy node only for testing purpose
model = torch.nn.Linear(10, 10)