AI-GrandChallenge / round-1

14 stars 11 forks source link

[MultiModel] tensorflow와 pytorch 모델을 같이 쓰고있습니다. #18

Open DeepLearningHB opened 4 years ago

DeepLearningHB commented 4 years ago

제 task에서 tensorflow 모델과 pytorch 모델을 같이 사용해서 결과를 얻는 방식으로 진행하고 있는데 어떤식으로 진행하면 좋을지 가이드라인 부탁드립니다.. bind_nsml 에서 어디를 건드려야할 지 잘 모르겠습니다..

bluebrush commented 4 years ago

두 가지를 동시에 사용하시려면, custom으로 docker image를 만드셔서 docker hub에 올리시고, setup.py에 해당 이미지의 링크를 가리키게 수정하셔야할 듯합니다.

DeepLearningHB commented 4 years ago

bind_nsml는 do not change라고 되어있는데 건드려도 무방한가요? 하나의 모델만 로드하게 되어있는거같습니다..