fix swag example for work with apex
This commit is contained in:
1 parent
0cf88ff084
commit
0f544625f4
3 files changed
+59
-78
No files matched your search
@@ -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
|
||||
```
|
||||
|
||||
|
||||
Reference in new issue
Block a user