diff --git a/examples/pytorch/multiple-choice/README.md b/examples/pytorch/multiple-choice/README.md index 735d1f5f33a..8d56ccfe3db 100644 --- a/examples/pytorch/multiple-choice/README.md +++ b/examples/pytorch/multiple-choice/README.md @@ -28,7 +28,7 @@ python examples/multiple-choice/run_swag.py \ --learning_rate 5e-5 \ --num_train_epochs 3 \ --output_dir /tmp/swag_base \ ---per_gpu_eval_batch_size=16 \ +--per_device_eval_batch_size=16 \ --per_device_train_batch_size=16 \ --overwrite_output ```