Nvidia Nemo의 ckpt파일 로드하기

공식문서에도 ckpt파일을 모델로 로드하는 방법이 나와있지 않고, 현시점 ChatGPT4한테 물어봐도 이상하게 알려준다.. 알고보니 상당히 간단한 방법으로 ckpt파일을 모델로 가져올 수 있었다.

ckpt을 모델로 가져오기

nemo.collections.asr.models.해당 model의 class명.load_from_checkpoint(체크포인트 파일명)

# ex) model = nemo.collections.asr.models.EncDecCTCModel.load_from_checkpoint("100epoch.ckpt")

load_from_checkpoint를 이용하여 가져올 수 있다.

.nemo로 저장하기

nemo 형식으로 저장하려면

model.save_to(output_path) # .nemo

를 해주면 된다.