文件名称：torchprofile:适用于PyTorch模型的通用且准确的MACs FLOPs分析器
文件大小：10KB
文件格式：ZIP
更新时间：2024-05-19 12:28:44
profiler pytorch Python
火炬简介 这是一个分析器，用于基于torch.jit.trace对PyTorch模型的MAC / FLOP进行torch.jit.trace 。 它比基于ONNX的事件探查器更通用，因为ONNX目前不支持PyTorch中的某些操作。 它比基于钩子的探查器更准确，因为它们无法探查torch.nn.Module操作。 安装 pip install torchprofile 入门 您应该首先定义您的PyTorch模型及其（虚拟）输入： import torch from torchvision . models import resnet18 model = resnet18 () inputs = torch . randn ( 1 , 3 , 224 , 224 ) 然后，您可以使用profile_macs测量MAC的数量： from torchprofile import pro
【文件预览】：
torchprofile-master
----torchprofile()
--------utils()
--------handlers.py(5KB)
--------profile.py(757B)
--------__init__.py(67B)
--------version.py(22B)
----examples()
--------profile_torchvision.py(537B)
--------profile_transformer.py(421B)
--------trace_linear.py(291B)
----LICENSE(1KB)
----setup.py(370B)
----README.md(876B)
----.gitignore(21B)