ColossalAI/tests/test_fx/test_pipeline at efdc240f1fba00efd670cea357b7f04ef976ed0c - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 15:29:48 +00:00

Files

History

Frank Lee 4f4d8c3656 [fx] added apex normalization to patched modules (#1300 )

* [fx] added apex normalization to patched modules

* remove unused imports

2022-07-14 14:24:13 +08:00

..

[fx] added apex normalization to patched modules (#1300 )

2022-07-14 14:24:13 +08:00

test_timm_model

[fx]add split module pass and unit test from pipeline passes (#1242 )

2022-07-12 13:45:01 +08:00

test_torchvision

[hotfix] torchvison fx unittests miss import pytest (#1277 )

2022-07-12 23:04:06 +08:00