ernie_gen执行训练中途卡死退出怎么办
收藏
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
收藏
请登录后评论
模型某些层的参数没读出来,不行再重新存下权重