📜  gpt2 简单恢复检查点 - Python 代码示例

📅  最后修改于: 2022-03-11 14:46:02.043000             🧑  作者: Mango

代码示例1
tf.reset_default_graph()
sess = gpt2.start_tf_sess()

gpt2.finetune(sess,
    file_name,
    model_name=model_name,
    checkpoint_dir=checkpoint_dir,
    run_name=run_name,
    steps=25,
)