From 3d5dea9bf0c33cc0e9952e1566f5c7097be72b0d Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 10 Feb 2022 14:52:07 +0100 Subject: [PATCH] Add example batch size to all commands (#15596) --- examples/pytorch/language-modeling/README.md | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/examples/pytorch/language-modeling/README.md b/examples/pytorch/language-modeling/README.md index c768f5ec31..00e3a87053 100644 --- a/examples/pytorch/language-modeling/README.md +++ b/examples/pytorch/language-modeling/README.md @@ -39,6 +39,8 @@ python run_clm.py \ --model_name_or_path gpt2 \ --dataset_name wikitext \ --dataset_config_name wikitext-2-raw-v1 \ + --per_device_train_batch_size 8 \ + --per_device_eval_batch_size 8 \ --do_train \ --do_eval \ --output_dir /tmp/test-clm @@ -54,6 +56,8 @@ python run_clm.py \ --model_name_or_path gpt2 \ --train_file path_to_train_file \ --validation_file path_to_validation_file \ + --per_device_train_batch_size 8 \ + --per_device_eval_batch_size 8 \ --do_train \ --do_eval \ --output_dir /tmp/test-clm @@ -83,6 +87,8 @@ python run_mlm.py \ --model_name_or_path roberta-base \ --dataset_name wikitext \ --dataset_config_name wikitext-2-raw-v1 \ + --per_device_train_batch_size 8 \ + --per_device_eval_batch_size 8 \ --do_train \ --do_eval \ --output_dir /tmp/test-mlm @@ -95,6 +101,8 @@ python run_mlm.py \ --model_name_or_path roberta-base \ --train_file path_to_train_file \ --validation_file path_to_validation_file \ + --per_device_train_batch_size 8 \ + --per_device_eval_batch_size 8 \ --do_train \ --do_eval \ --output_dir /tmp/test-mlm @@ -139,6 +147,8 @@ python run_plm.py \ --model_name_or_path=xlnet-base-cased \ --dataset_name wikitext \ --dataset_config_name wikitext-2-raw-v1 \ + --per_device_train_batch_size 8 \ + --per_device_eval_batch_size 8 \ --do_train \ --do_eval \ --output_dir /tmp/test-plm @@ -151,6 +161,8 @@ python run_plm.py \ --model_name_or_path=xlnet-base-cased \ --train_file path_to_train_file \ --validation_file path_to_validation_file \ + --per_device_train_batch_size 8 \ + --per_device_eval_batch_size 8 \ --do_train \ --do_eval \ --output_dir /tmp/test-plm