Tensorflow model restore시 주의점 & downgrading the version

텐서플로우로 학습시킨 모델을 저장한 후, 나중에 불러올 때(restore) 저장했던 환경과 불러오는 환경의 텐서플로우 버전이 다르면 에러가 발생할 수 있다. (이것 때문에 또 얼마나 헤멨었는지..)

버전에 맞추어 재인스톨하는 방법은 간단하다. (다운그레이드라도 –upgrade 옵션 사용)

만약 1.1.0으로 다운그레이드하고자 한다면,


pip install --upgrade tensorflow==1.1.0

 

Advertisements
Posted in Dev

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s