Tensorflow : tf.gather가 큰 배열에서 작동하지 않습니다.

Question

크기가 (29000,200,1024) (7Go) 인 내 numpy 배열에 문제가 있습니다. 내 데이터 세트의 이미지 기능.

일단로드되면, my 함수는 현재 배치를 텐서로 만들기 위해 인덱스를받습니다. 불행히도 다음을 사용 :

tf.gather(array, indices) 

동결. 예를 들어 배열 [0] 인쇄는 즉시 작동하지만. array_tensor (indice)를 직접 사용할 수 있도록 convert_to_tensor으로 숫자가 작은 배열을 변환하려고 시도했지만 다시 convert_to_tensor은 메모리 제한 오류를 발생시킵니다.

주위에 어떤 문제 해결?

는 2GB의 GraphDef 제한에 따라 GraphDef에 모든 것을, 인라인되도록, 연산 정의에 데이터를 포함 tf.constant 연산으로 변환 TF 연산 건설 API에 직접 NumPy와 배열을 전달하면

Answer 1

대단히 감사합니다.

이렇게하려면 var=tf.Variable(my_placeholder)을 만들고 var.initializer, feed_dict={my_placeholder: np_array}을 실행하여이 변수를 초기화하십시오. 이것은 numpy 배열 데이터를 변수 저장소에 직접 저장합니다.