diff --git a/README.md b/README.md index 4bd4345..2a78ac1 100644 --- a/README.md +++ b/README.md @@ -1,14 +1,6 @@ # torchtops(Pytorch TOPS) [![Pypi version](https://img.shields.io/pypi/v/torchtops.svg)](https://pypi.org/project/torchtops/) -This toolkit evaluate layer level TOPS(Terra Operations Per Second) - -these items will be retuned by profile -- layer_name_list (List[str]) - List of layer names -- latency_list (List[float]) - List of latency in miliseconds -- tops_list (List[float]) - List of tops -- modules (List[nn.Module]) - List of nn.Module -- total_flops (int) - total flops of the model ## Quickstart @@ -26,11 +18,11 @@ res = profile(model, img) res = filter_modules(res, target_modules=["Conv2d"]) # filter nn.Module you want to get -tops_list, layer_name_list, modules = zip( +tops_list, layer_name_list, module_list = zip( *sorted(zip(res["tops_list"], res["layer_name_list"], res["module_list"])) ) -for tops, layer_name, module in zip(tops_list, layer_name_list, modules): +for tops, layer_name, module in zip(tops_list, layer_name_list, module_list): print(f"{tops:.3f} => {layer_name} : {module}") ``` @@ -64,11 +56,5 @@ this results depend on your hardware python3.8 tools/example.py ``` -### ResNet152 - - -### Vit_b_16 - - -### Swin_v2_b - +### ResNet50 + diff --git a/resources/resnet152.jpg b/resources/resnet152.jpg deleted file mode 100644 index 9cb5142..0000000 Binary files a/resources/resnet152.jpg and /dev/null differ diff --git a/resources/swin_v2_b.jpg b/resources/swin_v2_b.jpg deleted file mode 100644 index f9e1b95..0000000 Binary files a/resources/swin_v2_b.jpg and /dev/null differ diff --git a/resources/vit_b_16.jpg b/resources/vit_b_16.jpg deleted file mode 100644 index 9a0de03..0000000 Binary files a/resources/vit_b_16.jpg and /dev/null differ diff --git a/torchtops/__init__.py b/torchtops/__init__.py index 99eda0b..1b1483a 100644 --- a/torchtops/__init__.py +++ b/torchtops/__init__.py @@ -1,4 +1,4 @@ from .layer_profiler import profile from .utils import filter_modules -__version__ = "0.1.1" +__version__ = "0.1.2"