tensorflow1.0到temsorflow2.0代码转换

  1. 现阶段很多github都是tf1.0的代码,那么我们如何在tf2.0环境下运行tf1.0的代码
    首先第一点将
import tensorflow as tf

改成

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
  1. 其次在导入LSTM或者Dropout等包的时候,需要将如下代码进行替换
tf.contrib.rnn.BasicLSTMCell

改成

tf.nn.rnn_cell.BasicLSTMCell

和将

tf.contrib.rnn.DropoutWrapper

改成

tf.nn.rnn_cell.DropoutWrapper

你可能感兴趣的:(tensorflow,深度学习,人工智能)