Metadata-Version: 2.4
Name: modelinfo
Version: 0.0.1
Summary: A Python library for model FLOPs and throughput benchmarking
Author: 17fine
Author-email: 17fine <2756601885@qq.com>
License: MIT
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: torch
Requires-Dist: torchvision
Requires-Dist: thop
Requires-Dist: torchinfo
Dynamic: author
Dynamic: requires-python

```python
from modelinfo import ModelProfiler
from torchvision.models import resnet18

model = resnet18()
input_size = (1, 3, 224, 224)

profiler = ModelProfiler(model, input_size)

profiler.summary()
profiler.flops_memory(precision='fp16')
profiler.benchmark(precision='fp16', training=True)

```
