fix swag example for work with apex

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

No files matched your search

+2 -2
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
```