首页 PaddleNLP 帖子详情
ernie_gen执行训练中途卡死退出怎么办
收藏
快速回复
PaddleNLP 问答数据训练NLP 509 1
ernie_gen执行训练中途卡死退出怎么办
收藏
快速回复
PaddleNLP 问答数据训练NLP 509 1

ernie_gen执行训练中途卡死退出

模型也没输出,突然就退出了,任何报错都没有

代码也贴上

import paddlehub as hub

module = hub.Module(name="ernie_gen")

result = module.finetune(
    train_path='./c.txt',
    save_dir='./ernie_output',
    #dev_path='dev.txt',
    batch_size=2,
    use_gpu=True,
    max_steps=400,
    max_encode_len=128,
    max_decode_len=128,
    noise_prob=0.1,
    log_interval=50,
    save_interval=2000
)

module.export(params_path=result['last_save_path'], module_name="ernie_gen_test", author="test")
0
收藏
回复
全部评论(1)
时间顺序
FutureSI
#2 回复于2021-11

模型某些层的参数没读出来,不行再重新存下权重

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户