1
내 신경망 모델은 GPU를 사용하여 Theano의 Keras에서 작성되었습니다. 향후 다른 컴퓨터에서 Pickle을 사용하여 저장하고 있습니다.GPU를 사용하지 않는 Keras의 신경망 모델 예측
GPU없이 예측을 위해 모델을 사용할 수 있습니까?
A
답변
2
확실히. 일반적인 유스 케이스입니다. GPU는 교육을 강화하는 데 도움이되지만 프로덕션에서는 사용할 수없는 경우가 있습니다 (예 : 고객의 전화로 실행하는 경우).
나는 Theano를 많이 알지 못하지만, 그들은 tensorflow.serving과 동등한 것을가집니다. 항상 훈련 된 Model 객체를 직렬화하여 다른 컴퓨터에서 읽을 수 있습니다.
- 내장
keras.models.save_model및 파일을 HDF5하는 모델을 덤프keras.models.load_model. - 피클을 필요로하거나 선호하는 경우 - 기본적으로 Keras는 지원하지 않지만이 속임수를 사용할 수 있습니다. - http://zachmoshe.com/2017/04/03/pickling-keras-models.html