- 现阶段很多github都是tf1.0的代码,那么我们如何在tf2.0环境下运行tf1.0的代码
首先第一点将
import tensorflow as tf
改成
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
- 其次在导入LSTM或者Dropout等包的时候,需要将如下代码进行替换
tf.contrib.rnn.BasicLSTMCell
改成
tf.nn.rnn_cell.BasicLSTMCell
和将
tf.contrib.rnn.DropoutWrapper
改成
tf.nn.rnn_cell.DropoutWrapper