如何存储损失函数的梯度w.r.t.训练期间给定训练批次的模型参数?

时间:2019-07-09 17:08:04

标签: python tensorflow deep-learning

我需要存储损失函数w.r.t的梯度值。训练过程中每个训练样本的模型参数。它可以采用batch_size * num_params形状的矩阵形式。每次反向传播后,是否有任何方法可以在每次迭代时保存这些值?

0 个答案:

没有答案