未验证 提交 1c30cff1 编写于 作者: 小湉湉's avatar 小湉湉 提交者: GitHub

fix gpus of ernie_sat, test=tts (#2355)

上级 efaa5400
......@@ -3,7 +3,7 @@
set -e
source path.sh
gpus=0,1
gpus=0,1,2,3,4,5,6,7
stage=0
stop_stage=100
......
......@@ -3,7 +3,7 @@
set -e
source path.sh
gpus=0,1
gpus=0,1,2,3,4,5,6,7
stage=0
stop_stage=100
......
......@@ -3,7 +3,7 @@
set -e
source path.sh
gpus=0,1
gpus=0,1,2,3,4,5,6,7
stage=0
stop_stage=100
......
......@@ -71,7 +71,7 @@ class ErnieSATSampler(BatchSampler):
assert isinstance(drop_last, bool), \
"drop_last should be a boolean number"
from paddle.fluid.dygraph.parallel import ParallelEnv
from paddle.distributed import ParallelEnv
if num_replicas is not None:
assert isinstance(num_replicas, int) and num_replicas > 0, \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册