不能通过tf.contrib.layers.recompute_grad训练重量

Question

当我在图层上使用tf.contrib.layers.recompute_grad时，可训练的权重将被消除。如何保存每个时期在向后传递中重新计算的权重？