[TensorFlow] tf 2.x에서 tf 1.x 코드 실행하기

RNN으로 예측 모델을 만드려고 찾아본 코드가 텐서플로우 1.x로 돼있는 게 많아서 구글링 결과로 나온 사이트에 안 들어가본 데가 없는 것 같다. 또 필요해질 때를 대비해서 내가 까먹지 않기 위해 정리해놓는다.

 

⭐ 현재 사용 버전 : TensorFlow 2.4.0

초기 설정

◽ import tensorflow.compat.v1 as tf

compatibility module인 tensorflow.compat 중에서 1.x 버전을 의미하는 v1 import

 

◽ tf.compat.v1.disable_eager_execution()

텐서플로우 2.0에서 기본으로 설정된 즉시 실행(execution eagerly) 끄기

 

모듈

TensorFlow 1.x	tensorflow.compat.v1
tf.contrib.rnn.BasicLSTMCell	tf.compat.v1.nn.rnn_cell.LSTMCell
tf.contrib.rnn.DropoutWrapper	tf.compat.v1.nn.rnn_cell.DropoutWrapper
tf.contrib.rnn.MultiRNNCell	tf.compat.v1.nn.rnn_cell.MultiRNNCell
tf.nn.dynamic_rnn	tf.compat.v1.nn.dynamic_rnn
tf.contrib.layers.fully_connected	tf.compat.v1.layers.dense

 

contrib 모듈은 2.x 버전에서 완전히 사라졌기 때문에 tensorflow.compat.v1을 import한다고 해도 지원이 되지 않는다. 위의 표에 적힌 코드는 "비슷한 역할을 하게 해주는" 코드라고 보면 된다.

 

 

 

찾다찾다 안돼서 다운그레이드할까 고민도 했는데 코드를 저렇게 바꾸고 나서 모델이 정상 작동해서 천만다행이었다 😊

 

 

🙏 틀린 내용이나 오타가 있다면 댓글로 편하게 남겨주세요!