관리 메뉴

블로그

[학습] PyTorch 토크나이저 저장 본문

공부

[학습] PyTorch 토크나이저 저장

beenu 2023. 10. 3. 14:02
반응형

모델과 토크나이저를 저장할 때는 이 둘을 같은 디렉터리에 저장하는 것이 관리하기 편리하며, 나중에 둘을 함께 로드할 때도 편리하기 때문에 같은 파일에 저장하는 것을 습관화 하는게 좋을 것 같다.

 

⬇️저장 방법

# 토크나이저 및 모델 저장

tokenizer.save_pretrained(save_directory)
model.save_pretrained(save_directory)
728x90
반응형