search

VinAIResearch / PhoBERT

PhoBERT: Pre-trained language models for Vietnamese (EMNLP-2020 Findings)
MIT License
636 stars 92 forks source link

Cách load trained model #18

Closed 17521121 closed 3 years ago

17521121 commented 3 years ago

Chào anh, lúc trước để load model đã train thì em load bằng self.roberta = RobertaModel(config) thay vì self.roberta = RobertaModel.from_pretrained("PhoBERT_base_transformers/model.bin", config=config) (Em bọc trong class riêng ) Và model em save bằng torch.save (state_dict), Vấn đề là em giờ k biết file config đó nằm đâu, hoặc có cách nào khác để load model đã train , mong anh giải đáp

datquocnguyen commented 3 years ago

Hi, I have no idea to be able to answer your question :|