fix swag example for work with apex

This commit is contained in:
thomwolf
2018-12-13 13:35:59 +01:00
parent 0cf88ff084
commit 0f544625f4
3 changed files with 59 additions and 78 deletions

View File

@@ -442,12 +442,12 @@ python run_swag.py \
--do_train \
--do_lower_case \
--do_eval \
--data_dir $SWAG_DIR/data
--data_dir $SWAG_DIR/data \
--train_batch_size 16 \
--learning_rate 2e-5 \
--num_train_epochs 3.0 \
--max_seq_length 80 \
--output_dir /tmp/swag_output/
--output_dir /tmp/swag_output/ \
--gradient_accumulation_steps 4
```