바뀜

305 바이트 추가됨 ,  2022년 4월 12일 (화) 17:46
50번째 줄: 50번째 줄:  
|훈련 전에 다음의 코드를 넣는다.<syntaxhighlight lang="python">
 
|훈련 전에 다음의 코드를 넣는다.<syntaxhighlight lang="python">
 
config = tf.ConfigProto()
 
config = tf.ConfigProto()
 +
# 2버전 부터는 경로가 바뀌었기 때문에 다음과 같이 넣자. tf.compat.v1.ConfigProto()
 
config.gpu_options.allow_growth = True
 
config.gpu_options.allow_growth = True
 
session = tf.Session(config=config)
 
session = tf.Session(config=config)
</syntaxhighlight>
+
# 2버전 부터는 경로가 바뀌었기 때문에 다음의 경로를 지정하자. tf.compat.v1.Session
 +
</syntaxhighlight>텐서플로우 2부턴 경로가 바뀌었기 때문에 config에 다음과 같이 넣자.
 
|-
 
|-
 
|메모리 할당 비율을 설정
 
|메모리 할당 비율을 설정