데이터과학 삼학년

[Transformers] 파이토치 가중치만 있는 모델을 TF로 사용?! 본문

Natural Language Processing

[Transformers] 파이토치 가중치만 있는 모델을 TF로 사용?!

Dan-k 2023. 9. 11. 09:00
반응형

- huggingface에서 모델을 임포트하여 TF로 사용하려 할때!!

  ㄴ 파이토치 가중치만 존재하는 모델이 있다.

 

- 이때 tf에서 사용할 수 있는 방법은!!!! 존재한다. 바로...

  ㄴ from_pt=True

from transfomers import TFAutoModel

model_ckt = 'distilbert-base-uncased'
tf_model = TFAutoModel.from_pretrained(model_ckt)


## 파이토치 가중치만 있는 모델
model_ckt_torch = 'xlm-roberta-base'
tf_xlmr = TFAutoModel.from_pretrained(model_ckt_torch, from_pt=True)
-> 자동으로 파이토치 가중치를 다운로드해 변환

 

 

728x90
반응형
LIST
Comments