主要介绍了可视化pytorch 模型中不同BN层的running mean曲线实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

加载模型字典 逐一判断每一层，如果该层是bn 的 running mean，就取出参数并取平均作为该层的代表 对保存的每个BN层的数值进行曲线可视化 from functools import partial import pickle import torch import matplotlib.pyplot as plt pth_path = 'checkpoint.pth' pickle.load = partial(pickle.load, encoding="latin1") pic