文件名称：torchprof:PyTorch逐层模型分析器
文件大小：14KB
文件格式：ZIP
更新时间：2024-05-31 21:23:46
Python
火炬教授 一个用于PyTorch模型的逐层分析的最小依赖库。 所有度量标准都是使用PyTorch autograd profiler导出的。 快速开始 pip install torchprof import torch import torchvision import torchprof model = torchvision . models . alexnet ( pretrained = False ). cuda () x = torch . rand ([ 1 , 3 , 224 , 224 ]). cuda () # `profile_memory` was added in PyTorch 1.6, this will output a runtime warning if unsupported. with torchprof . Profile ( model
【文件预览】：
torchprof-master
----.flake8(107B)
----.circleci()
--------config.yml(437B)
----requirements.txt(32B)
----torchprof()
--------profile.py(5KB)
--------__init__.py(104B)
--------display.py(9KB)
----LICENSE(1KB)
----setup.py(1KB)
----README.md(15KB)
----Makefile(156B)
----tests()
--------test_set_paths.py(576B)
--------test_profile.py(10KB)
--------__init__.py(0B)
----.gitignore(2KB)