update

bitallin · Jul 21, 2022 · 379b462 · 379b462
1 parent 713739d
commit 379b462
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -96,7 +96,7 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 python -m src.train \
 --early_stop_times "20" \
 --warmup_steps "-1" \
 --dev_data_ratio "0.01" \
---training_mode "normal" \
+--training_mode "ddp" \
 --amp true \
 --freeze_embedding false
 ```

diff --git a/command/train.sh b/command/train.sh
@@ -12,7 +12,7 @@ cd .. && CUDA_VISIBLE_DEVICES=0,1,2,3 python -m src.train \
 --early_stop_times "20" \
 --warmup_steps "-1" \
 --dev_data_ratio "0.01" \
---training_mode "normal" \
+--training_mode "ddp" \
 --amp true \
 --freeze_embedding false