如何观察PyTorch模型中不同BN层的running mean曲线实例？

本文共计677个文字，预计阅读时间需要3分钟。

加载模型字典，逐层判断，若该层是bn的running mean，则取出参数并计算平均值作为该层的代表。对每个BN层的数值进行曲线可视化。