AI-GrandChallenge / round-1

14 stars 11 forks source link

[tr-1] multiple models 사용시 제출 방법 문의 #101

Open G-Bong opened 4 years ago

G-Bong commented 4 years ago

여러가지 모델을 사용하여 tr-1 문제를 해결하려고 계획 중입니다.

nsml submit 할 때 세션에 장되어잇는 델을 제출해야 하는거 같은데요, multiple models로 구성된 코드셋에 대해 어떻게 run하고 제출하면 좋을지 방법 알려주시면 감사하겠습니다.

nsml-admin commented 4 years ago

안녕하세요.

nsml.save()할때, nsml.bind()에 bind된 save함수가 호출되는데요 (https://github.com/AI-GrandChallenge/round-1/blob/master/iitp_falling/main.py#L152)

bind된 save함수의 첫번째 argument로 directory가 들어오는데 그쪽에 저장할 모델들을 저장하면 됩니다.

마찬가지로 submit할때는 paused부분에서 nsml.bind에 bind된 load가 호출되는데요, load함수의 첫번째 argument로 save에 저장했던 폴더가 그대로 들어옵니다.

load에서 저장했던 모델들을 load하고, nsml.bind에 bind된 infer함수에서 inference를 진행하면 될것같습니다.

G-Bong commented 4 years ago

@nsml-admin 답변 감사합니다. 일단 해봐야 겠네요.

혹시 docker image 내에 모델을 저장하고 순차적으로 모델을 load 및 사용하는 것도 가능할까요?

nsml-admin commented 4 years ago

@G-Bong 네, 이미지내에 모델저장, 코드내에서 해당경로의 모델을 불러오는 경우도 가능합니다.

하지만 이미지 사이즈가 너무커지면 실행하는데 오래걸리고, 서버에 무리가 갈만큼 큰이미지라면 저희가 임의로 삭제할수도있습니다.