fix gpus of ernie_sat, test=tts

pull/2355/head
TianYuan 3 years ago
parent efaa5400e8
commit fdc8b25905

@ -3,7 +3,7 @@
set -e set -e
source path.sh source path.sh
gpus=0,1 gpus=0,1,2,3,4,5,6,7
stage=0 stage=0
stop_stage=100 stop_stage=100

@ -3,7 +3,7 @@
set -e set -e
source path.sh source path.sh
gpus=0,1 gpus=0,1,2,3,4,5,6,7
stage=0 stage=0
stop_stage=100 stop_stage=100

@ -3,7 +3,7 @@
set -e set -e
source path.sh source path.sh
gpus=0,1 gpus=0,1,2,3,4,5,6,7
stage=0 stage=0
stop_stage=100 stop_stage=100

@ -71,7 +71,7 @@ class ErnieSATSampler(BatchSampler):
assert isinstance(drop_last, bool), \ assert isinstance(drop_last, bool), \
"drop_last should be a boolean number" "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: if num_replicas is not None:
assert isinstance(num_replicas, int) and num_replicas > 0, \ assert isinstance(num_replicas, int) and num_replicas > 0, \

Loading…
Cancel
Save