[TIL]KDT팀프로젝트2021/03/17

less than 1 minute read

torchserve를 담당하시는 팀원분과 어떻게 모델 훈련 결과를 넘겨받을 수 있는지 얘기했다. 내가 모델 학습시킬때 쓴 pth, json 파일 불러온 상태에서 cfg.dump로 .yaml에 찍히는 내용 텍스트파일로 만든 뒤 torchserve로 넘겨주면 된다.
중간발표를 위해 model training과 torchserve로 연계되는 부분 팀원분과 같이 상의해서 ppt 페이지 만들었다.
어제 코랩 런타임 끊겼었다가 복구된 이후 30000, 40000iteration 돌리고 지금 50000iteration 돌리는 중이다. 학습 후 pth와 json, last checkpoint 저장 중이다.

Categories:

Updated: