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, )