提交 c1762638 编写于 作者: W wangxiao

update run.py of 3 demos

上级 991e9071
......@@ -2,6 +2,6 @@ import paddlepalm as palm
if __name__ == '__main__':
controller = palm.Controller('config.yaml')
controller.load_pretrain('../../pretrain_model/bert/params')
controller.load_pretrain('../../pretrain/bert-en-uncased-large/params')
controller.train()
......@@ -2,7 +2,7 @@ import paddlepalm as palm
if __name__ == '__main__':
controller = palm.Controller('config.yaml', task_dir='tasks')
controller.load_pretrain('../../pretrain_model/ernie/params')
controller.load_pretrain('../../pretrain/ernie-en-uncased-large/params')
controller.train()
controller = palm.Controller(config='config.yaml', task_dir='tasks', for_train=False)
......
......@@ -2,6 +2,6 @@ import paddlepalm as palm
if __name__ == '__main__':
controller = palm.Controller('config.yaml', task_dir='tasks')
controller.load_pretrain('../../pretrain_model/ernie/params')
controller.load_pretrain('../../pretrain/ernie-en-uncased-large/params')
controller.train()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册