github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-18 07:31:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx] fixed compatiblity issue with torch 1.10 (#1331)

Browse Source
This commit is contained in:
Frank Lee
2022-07-18 11:41:27 +08:00
committed by GitHub
parent 069d6fdc84
commit 75abc75c15
7 changed files with 32 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_fx/test_tracer/test_timm_model/test_timm_model.py
View File

@@ -1,9 +1,5 @@
import torch
import pytest
try:
import timm.models as tm
except:
pass
import timm.models as tm
from colossalai.fx import ColoTracer
from torch.fx import GraphModule