如何载入保存训练好的模型？ about paddlenlp HOT 4 CLOSED

和model.save_pretrained('./checkpoint')及tokenizer.save_pretrained('./checkpoint')对应，使用model.from_pretrained('./checkpoint')及tokenizer.from_pretrained('./checkpoint')就可以了

from paddlenlp.

model.from_pretrained和tokenizer.from_pretrained是支持按照bert-base-uncased这种模型名称或者本地训练模型的路径来导入模型的

from paddlenlp.

另外通过

model.save_pretrained('./checkpoint')
tokenizer.save_pretrained('./checkpoint')

保存下来的内容（如下）中的model_state.pdparams是通过Paddle原生的save接口保存的，也可以使用Paddle原生的load接口来加载

-rw-r--r-- 1 root root  416 Mar  2 07:56 model_config.json
-rw-r--r-- 1 root root 632M Mar  2 07:56 model_state.pdparams
-rw-r--r-- 1 root root  141 Mar  2 07:56 tokenizer_config.json
-rw-r--r-- 1 root root 227K Mar  2 07:56 vocab.txt

from paddlenlp.

另外通过

model.save_pretrained('./checkpoint')
tokenizer.save_pretrained('./checkpoint')

保存下来的内容（如下）中的model_state.pdparams是通过Paddle原生的save接口保存的，也可以使用Paddle原生的load接口来加载

-rw-r--r-- 1 root root  416 Mar  2 07:56 model_config.json
-rw-r--r-- 1 root root 632M Mar  2 07:56 model_state.pdparams
-rw-r--r-- 1 root root  141 Mar  2 07:56 tokenizer_config.json
-rw-r--r-- 1 root root 227K Mar  2 07:56 vocab.txt

谢谢！

from paddlenlp.