From 8607233679ad549641157a6ca4f90e4c0d42676b Mon Sep 17 00:00:00 2001 From: Ben Johnson Date: Wed, 20 Feb 2019 13:58:54 -0500 Subject: [PATCH] Update run_openai_gpt.py --- examples/run_openai_gpt.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/examples/run_openai_gpt.py b/examples/run_openai_gpt.py index 7a434ceaca..546c11b528 100644 --- a/examples/run_openai_gpt.py +++ b/examples/run_openai_gpt.py @@ -17,7 +17,15 @@ Adapted from https://github.com/huggingface/pytorch-openai-transformer-lm/blob/master/train.py It self adapted from https://github.com/openai/finetune-transformer-lm/blob/master/train.py - This script with default values fine-tunes and evaluate a pretrained OpenAI GPT on the RocStories dataset + This script with default values fine-tunes and evaluate a pretrained OpenAI GPT on the RocStories dataset: + python run_openai_gpt.py \ + --model_name openai-gpt \ + --do_train \ + --do_eval \ + --train_dataset $ROC_STORIES_DIR/cloze_test_val__spring2016\ -\ cloze_test_ALL_val.csv \ + --eval_dataset $ROC_STORIES_DIR/cloze_test_test__spring2016\ -\ cloze_test_ALL_test.csv \ + --output_dir ../log \ + --train_batch_size 16 \ """ import argparse import os