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

PaddlePaddle · Sep 6, 2022 · 1c30cff · 1c30cff
1 parent efaa540
commit 1c30cff
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/examples/aishell3/ernie_sat/run.sh b/examples/aishell3/ernie_sat/run.sh
@@ -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
 

diff --git a/examples/aishell3_vctk/ernie_sat/run.sh b/examples/aishell3_vctk/ernie_sat/run.sh
@@ -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
 

diff --git a/examples/vctk/ernie_sat/run.sh b/examples/vctk/ernie_sat/run.sh
@@ -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
 

diff --git a/paddlespeech/t2s/datasets/sampler.py b/paddlespeech/t2s/datasets/sampler.py
@@ -71,7 +71,7 @@ def __init__(self,
         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, \