Figure1.作图疑问

[Hardcandies]

Hi, 麻烦问下。

对Fig.1的梯度均值，统计的是整体loss对模型的梯度向量的L2范数吗？ 对Fig.1的梯度方差，统计的是[正样本doc的loss的梯度的L2范数, 负样本doc的loss梯度的L2范数】这个向量的方差吗？

如果可以的话，可以提供一个简单的作图代码。感谢！