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,