You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/fx
Frank Lee 11973d892d
[fx] added torchvision model tracing testing (#1216)
2 years ago
..
passes [fx] temporarily used (#1215) 2 years ago
tracer [fx] added torchvision model tracing testing (#1216) 2 years ago
__init__.py [fx] supported data-dependent control flow in model tracing (#1185) 2 years ago
proxy.py [fx] added testing for all albert variants (#1211) 2 years ago