machine learning

tensorflow 1 -> tensorflow 2 변환

단창 2020. 10. 30. 11:46

기본적으로 tf2에서 tf1코드를 돌릴수 있음 

import tensorflow.compat.v1 as tf
tf
.disable_v2_behavior()

하지만 이렇게 하면 tf2 의 기능들을 사용할수 없음 

참고로 tf.keras 을 이용해 고수준의 코드를 짯으면 바꿀것도 없음 

reference ) 

www.tensorflow.org/guide/migrate?hl=ko

 

텐서플로 1 코드를 텐서플로 2로 바꾸기  |  TensorFlow Core

Note: 이 문서는 텐서플로 커뮤니티에서 번역했습니다. 커뮤니티 번역 활동의 특성상 정확한 번역과 최신 내용을 반영하기 위해 노력함에도 불구하고 공식 영문 문서의 내용과 일치하지 않을 수

www.tensorflow.org

 

그냥 앞으로 무조건 tf.keras와 pytorch만 쓰자

반응형