Check Torch Model Structure, Flops, and Params| 东毅居士

Check Torch Model Structure, Flops, and Params

作者：XD / 发表： 2023年5月6日 05:49 / 更新： 2023年5月6日 05:50 / 编程笔记 / 阅读量：1014

Check Torch Model Structure, Flops, and Params

from torchinfo import summary
images.tensors.size() # (3, 640, 640)
summary(torch_model, images.tensors.size())

from thop import profile
flops, params = profile(torch_model, inputs=[images], verbose=False)

from ptflops import get_model_complexity_info
flops, params = get_model_complexity_info(torch_model, (3, 640, 640), as_strings=True, print_per_layer_stat=True)
flops, params = get_model_complexity_info(torch_model, [images.tensors.size()], as_strings=True, print_per_layer_stat=True)

本文作者：XD 转载请标明出处：http://www.eadst.com/blog/185

本站采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。

上一篇
阿里云域名迁移至腾讯云

下一篇
Python: Load and Save A JSON File

Category

标签云

OpenAI Hilton Heatmap Hungarian BeautifulSoup XGBoost LaTeX BF16 Zip SQLite AI OCR 搞笑 logger LeetCode WebCrawler Review Docker Paper Cloudreve GGML LLAMA scipy Streamlit torchinfo PyCharm VPN HuggingFace LoRA Nginx tar FP16 Interview VGG-16 ModelScope FP64 Qwen PIP CV Windows Translation Google Domain Math Linux HaggingFace Conda Bipartite NLTK InvalidArgumentError git-lfs Paddle MD5 Random Video JSON UNIX COCO VSCode GoogLeNet Tracking Vim Pandas LLM Git 签证 Quantization Michelin 证件照 Jetson Python Tensor Color Django ResNet-50 Base64 Distillation BTC Pickle CSV OpenCV CAM v2ray Algorithm GIT uWSGI transformers 报税 Gemma diffusers PDF SPIE Disk FP32 C++ Pillow PyTorch llama.cpp Web Quantize Food Plate CUDA Land Vmess YOLO Statistics Data Pytorch FlashAttention TSV Plotly Sklearn Magnet Safetensors Transformers ChatGPT 算法题 Dataset hf Use Markdown Baidu Bert Knowledge Proxy Tiktoken Numpy CTC Permission DeepStream 阿里云 Anaconda Augmentation GPTQ Password SVR TensorRT Diagram Excel Shortcut ONNX NLP Logo 腾讯云 Crawler Input Datetime SQL TensorFlow Hotel QWEN Github Bitcoin GPT4 Qwen2.5 Card NameSilo FastAPI Bin Ubuntu git 域名 Ptyhon XML CEIR Template EXCEL Attention Website RGB uwsgi Clash Image2Text 关于博主

站点统计

本站现有博文269篇,共被浏览446237次

本站已经建立2028天!

原 Check Torch Model Structure, Flops, and Params

作者：XD / 发表： 2023年5月6日 05:49 / 更新： 2023年5月6日 05:50 / 编程笔记 / 阅读量：1014

Check Torch Model Structure, Flops, and Params