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)
[](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"