TensorFlow 2.0中的tf.contrib.layers.recompute_grad在哪里？

Question

TensorFlow 2.0是否支持将计算作为“渐变过程中的重新计算”以节省内存？ TensorFlow 1.x具有tf.contrib.layers.recompute_grad，但是contrib在TF 2.0中已消失，看起来好像没有人搬过recompute_grad。

Answer 1

是的，他们没有移动它。这是Google的重大错误。因此，我不会移至Tensorflow 2。即使不是很重要，重新计算渐变并创建内存有效的图形也非常有用。