未验证 提交 5e1f7310 编写于 作者: X Xing Wu 提交者: GitHub

fix parallel error for 1.7 (#4243)

* fix parallel error for 1.7

* remove modification to run.sh
上级 22b8805b
......@@ -4,7 +4,7 @@ export FLAGS_eager_delete_tensor_gb=0.0
export FLAGS_fast_eager_deletion_mode=1
export CUDA_VISIBLE_DEVICES=0,1,2,3
python -m paddle.distributed.launch --selected_gpus=0,1,2,3 train.py \
python -m paddle.distributed.launch --selected_gpus=0,1,2,3 train.py \
--train_data ./data/train.tsv \
--test_data ./data/test.tsv \
--model_save_dir ./padding_models \
......
......@@ -330,7 +330,7 @@ class lex_net(fluid.dygraph.Layer):
self.linear_chain_crf = Linear_chain_crf(
param_attr=fluid.ParamAttr(
name='crfw', learning_rate=self.crf_lr),
name='linear_chain_crfw', learning_rate=self.crf_lr),
size=self.num_labels)
self.crf_decoding = Crf_decoding(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册