1
우리는 TOSORFlow 라이브러리를 사용하여 이미지 분류를 수행하는 iOS 앱을 제작하고 있습니다.TensorFlow iOS 메모리 경고
우리의 기계 학습 모델 (91MB, 400 클래스)과 TensorFlow의 '간단한'예제를 사용하여 1GB RAM을 가진 모든 iOS 장치에서 메모리 경고를받습니다. 2GB 모델에서는 아무런 경고 메시지가 표시되지 않지만 < 1GB 모델에서는 메모리가 완전히 소모되어 앱이 다운됩니다.
this iOS memory performance commit을 포함하는 마스터 분기의 최신 TensorFlow 코드를 사용하고 있지만 도움이되지는 않을 것으로 생각됩니다.
우리는 또한 set_allow_growth(true)과 set_per_process_gpu_memory_fraction()을 포함하여 우리의 TF 세션 객체에 다양한 GPU 옵션을 설정해 보았습니다. TF '간단한'예제 코드에
우리의 유일한 변화는 wanted_width과 299의 wanted_height 및 input_mean이로 128
가지고 다른 사람 실행의 input_std입니까? 우리 모델이 단순히 너무 큽니까?
나는이 질문을 결코 따라하지 않았지만, 나는 mmapped 모델을 사용하여 끝냈다. 그리고 그것은 그 문제를 해결했다. 응답 해 주셔서 감사합니다. – azdev