PyTorch常用小工具-Tensorboard、Summary、Profiler

  • 1 PyTorch可视化工具-Tensorboard
  • 2 summary计算每层参数个数
    • 2.1 安装与使用
    • 2.2 实战
  • 3 profiler 分析每个操作在GPU和CPU的时间花销
    • 3.1 教程
    • 3.2 实战 统计前向运行时间
  • 4 eval.py 和eval_2.py 对比
  • 5 工程文件、数据集、源码下载

1 PyTorch可视化工具-Tensorboard

PyTorch可视化工具-Tensorboard教程

2 summary计算每层参数个数

2.1 安装与使用

pip install torchsummary
#使用
from torchsummary import summary
summary(your_model, input_size=(channels, H<

你可能感兴趣的:(深度学习,论文格式必会的技能,传统/深度三维重建MVS系列,pytorch,深度学习,计算机视觉)