[일반] transformers 모델을 tf keras로 저장 어케해?

[Sesiwo(visiopo44)]

transformers모델을

tf keras의model을 오버라이딩해서 사용자 정의 레이어를 만들고 finetuning했는데 model.save로 저장하고 불러오려니 버그 생기네...;; 코랩리소스 1/5짜린데ㅠ

어떻게 저장하고 불러와야할까?ㅠ

- dc official App

[익명(110.70)]

함수형이나 시퀀셜 안썻으면 모델 웨이트만 뽑아와서 다시쓸땐 모델 그대로 정의하고 웨이트 불러와야 할걸요?

[Sesiwo(visiopo44)]

함수형 썼어요 - dc App

[익명(118.235)]

무슨 버그

[Sesiwo(visiopo44)]

https://stackoverflow.com/questions/76541629/how-to-save-and-load-transformer-model-fine-tuned-with-tfkeras
- dc App

[Sesiwo(visiopo44)]

그대로 로드했는데 인수가 예상되는것보다 많다는 듯한 내용? - dc App

[익명(118.235)]

뭘로 저장함 - dc App

[Sesiwo(visiopo44)]

save - dc App

[익명(118.235)]

h5같은 거 말임 - dc App

[Sesiwo(visiopo44)]

아 케라스 save임 save_weight아니고. 확장자 없이 저장함 - dc App