search

Naver-AI-Hackathon / AI-Vision

67 stars 34 forks source link

Tensorflow 용 bindmodel 및 infer 함수 #245

Open heasungkim opened 5 years ago

heasungkim commented 5 years ago

요청사항 tensorflow 를 이용하여 사용할 수 있는 bindmodel (save, load, infer) 베이스라인 함수를 제공해 주실 수 있나요?

nsml 에서 infer 가 실행되는 정확한 process를 알지 못하여 tensorflow infer를 코딩하는데 어려움이 있습니다. Keras 를 이용한 infer는 베이스라인이 존재하여 그 process를 알지 못해도 실행할 수 있었는데, tensorflow로 하려니 정보가 없어 어렵네요.

모델을 일단 1.정의 하고 2. weight를 load 하고, 3. infer 를 실행하는것이 기본일텐데, bind_model 하나의 함수만을 실행하는것은 아닐테고, 저희가 submit을 할 경우 원래 올렸던 코드에서 nsml.bind 가 특별히 동작한다던지 하는 과정을 알 수 없어 짜기가 힘듭니다.

간단한 코드라도 알려주시면 감사하겠습니다.

nsml-admin commented 5 years ago

안녕하세요.

다음 은 작년 해커톤때 사용했던 tensorflow예제인데 참고해보시기 바랍니다.

말씀해주신대로 submit할때의 순서는 다음과같습니다.

  1. 모델정의
  2. bind_model 에 정의된 모델을 argument로 입력
  3. nsml.bind()에 바인딩된 load 함수 실행. (weight는 이시점에 로드됩니다)
  4. nsml.bind()에 바인딩된 infer함수 실행

도움이 되었으면 좋겠습니다.

감사합니다.