"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3726754a6c646adcf9cb2135ab7f72dffe074473"
Unverified Commit 97c815da authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #305 from bkj/patch-1

Update run_openai_gpt.py
parents f50b82af 86072336
...@@ -17,7 +17,15 @@ ...@@ -17,7 +17,15 @@
Adapted from https://github.com/huggingface/pytorch-openai-transformer-lm/blob/master/train.py 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 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 argparse
import os import os
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment