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 / 编程笔记 / 阅读量：1995

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

标签云

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

站点统计

本站现有博文312篇,共被浏览744653次

本站已经建立2388天!

原 Check Torch Model Structure, Flops, and Params

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

Check Torch Model Structure, Flops, and Params