Skip to content

Graph dump of torchbench models, huggingface models, and TIMM models.

Notifications You must be signed in to change notification settings

yushangdi/fx_module_dumps

Repository files navigation

fx_module_dumps

Graph dump of torchbench models, huggingface models, and TIMM models.

Unzip the files

unzip huggingface_graphs.zip
unzip timm_graphs.zip
unzip torch_bench_graphs.zip

After unzipping the files, you can get start with example.py. It generate inputs for each graph and runs the graph with the generated input. There are also some example analysis you can do on the graphs.

About

Graph dump of torchbench models, huggingface models, and TIMM models.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published