search

Naver-AI-Hackathon / AI-Speech

31 stars 5 forks source link

nsml.save 시 summary 생성 #73

Open angrypark opened 4 years ago

angrypark commented 4 years ago

NSML을 좀더 효율적으로 사용하기 위해 NSML Documendation을 읽고 있는데요. 유용하다고 생각된 기능이 nsml model ls [SESSION] 시에 Summary에 해당 모델의 정보가 담긴다는 점이었습니다.

$ nsml model ls nsmlteam/mnist/4
Checkpoint    Last Modified    Elapsed    Summary                           Size
------------  ---------------  ---------  --------------------------------  ---------
0             32 minutes ago   3.785      epoch=0, loss=7.015200052175436   366.74 MB
1             31 minutes ago   25.032     epoch=1, loss=6.762363957929182   366.74 MB
2             31 minutes ago   24.942     epoch=2, loss=6.377185855899845   366.74 MB
3             30 minutes ago   25.053     epoch=3, loss=5.742813685992816   366.74 MB
4             30 minutes ago   24.971     epoch=4, loss=4.847646936640009   366.74 MB
11            29 minutes ago   25.032     epoch=1, loss=6.762363957929182   366.74 MB
21            29 minutes ago   24.942     epoch=2, loss=6.377185855899845   366.74 MB
31            28 minutes ago   25.053     epoch=3, loss=5.742813685992816   366.74 MB
41            28 minutes ago   24.971     epoch=4, loss=4.847646936640009   366.74 MB

그런데 nsml.save의 인자들을 살펴봐도 어떻게 summary를 지정하는 지에 대한 내용이 없어서요. 어떻게 위의 결과처럼 summary 정보를 저장할 수 있는지 궁금합니다.

nsml-admin commented 4 years ago

안녕하세요.

nsml.save()하기 직전에 nsml.report()에 기록된정보를 토대로 metadata를 저장하고 보여주게됩니다.

그점을 참고하시고 진행하시면 될것같습니다.

관련된 내용이 문서에 없는점 양해부탁드리며 조만간 보강하도록 하겠습니다.

감사합니다.

angrypark commented 4 years ago

앗 그렇군요! 감사합니다 잘 사용해보겠습니다 ㅎㅎ