From fdc8b2590543120b334da960c423885173e79247 Mon Sep 17 00:00:00 2001
From: TianYuan <white-sky@qq.com>
Date: Tue, 6 Sep 2022 10:36:25 +0000
Subject: [PATCH] fix gpus of ernie_sat, test=tts

---
 examples/aishell3/ernie_sat/run.sh      | 2 +-
 examples/aishell3_vctk/ernie_sat/run.sh | 2 +-
 examples/vctk/ernie_sat/run.sh          | 2 +-
 paddlespeech/t2s/datasets/sampler.py    | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/aishell3/ernie_sat/run.sh b/examples/aishell3/ernie_sat/run.sh
index cb354de41..d3efefe0c 100755
--- 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
index 5509fc4ad..8cd9d8d1b 100755
--- 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
index 94d130d41..cb80d1ad8 100755
--- 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
index 3c97d1dc4..cbc9764c5 100644
--- a/paddlespeech/t2s/datasets/sampler.py
+++ b/paddlespeech/t2s/datasets/sampler.py
@@ -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, \