文件名称：torchprof - 用于Pytorch模型的逐层分析的最小依赖库-python
文件大小：14KB
文件格式：ZIP
更新时间：2024-07-11 03:48:03
机器学习
torchprof - 用于Pytorch模型的逐层分析的最小依赖库 torchprof 用于 Pytorch 模型的逐层分析的最小依赖库。 所有指标都是使用 PyTorch autograd 分析器得出的。 快速入门 pip install torchprof import torch import torchvision import torchprof model = torchvision.models.alexnet(pretrained=False).cuda() x = torch.rand([1, 3, 224, 224]).cuda() with torchprof.Profile (model, use_cuda=True) as prof: model(x) print(prof.display(show_events=False)) # 等价于 `print(prof)` 和 `print(prof.display())` 模块 | 自 CPU 总量 | CPU总量| CUDA 总数 | 发生 ---------------|------------
【文件预览】：
torchprof-master
----.circleci()
--------config.yml(437B)
----.flake8(107B)
----.gitignore(2KB)
----README.md(15KB)
----torchprof()
--------profile.py(5KB)
--------__init__.py(104B)
--------display.py(9KB)
----tests()
--------test_profile.py(10KB)
--------test_set_paths.py(576B)
--------__init__.py(0B)
----LICENSE(1KB)
----requirements.txt(32B)
----setup.py(1KB)
----Makefile(156B)