[英]Migrate tf.contrib.layers.batch_norm to Tensorflow 2.0
[英]Where is tf.contrib.layers.recompute_grad in TensorFlow 2.0?
TensorFlow 2.0是否支持将计算作为“渐变过程中的重新计算”以节省内存? TensorFlow 1.x具有tf.contrib.layers.recompute_grad ,但是contrib
在TF 2.0中已经消失了,看起来好像没有人搬过 recompute_grad
。
是的,他们没有移动它。 这是Google的重大错误。 因此,我不会移至Tensorflow 2。 重新计算梯度并创建内存有效的图形非常有用,即使不是很重要。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.