New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[fx] added torchvision model tracing testing #1216

Merged

YuliangLiu0306 merged 2 commits into hpcaitech:main from FrankLeeeee:feature/torchvision-fx-testing

Jul 6, 2022

Contributor

FrankLeeeee commented Jul 6, 2022

In this PR, the following tasks are completed

ColoTracer is tested with models in torchvision
Fall back to original torch fx tracer when no meta arguments are given
In PyTorch 1.11, there is meta tensor support for pooling, thus added relevant patches

FrankLeeeee added 2 commits

July 6, 2022 17:32


          [fx] added torchvision model tracing testing

f0dffc5


          remove unused imports

7dbc919

YuliangLiu0306 approved these changes

View reviewed changes

YuliangLiu0306 merged commit 11973d8 into hpcaitech:main

FrankLeeeee deleted the feature/torchvision-fx-testing branch

January 26, 2023 07:08

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment