fix trianer

pull/2372/head
tianhao zhang 3 years ago
parent ed16f96a9c
commit 82e04d7815

@ -2,8 +2,8 @@
source path.sh
set -e
gpus=0,1,2,3
stage=0
gpus=1
stage=1
stop_stage=50
conf_path=conf/conformer.yaml
ips= #xx.xx.xx.xx,xx.xx.xx.xx

@ -19,6 +19,9 @@ from pathlib import Path
import paddle
from paddle import distributed as dist
import pdb
pdb.set_trace()
dist.init_parallel_env()
from visualdl import LogWriter
from paddlespeech.s2t.training.reporter import ObsScope
@ -176,7 +179,7 @@ class Trainer():
def init_parallel(self):
"""Init environment for multiprocess training.
"""
dist.init_parallel_env()
# dist.init_parallel_env()
@mp_tools.rank_zero_only
def save(self, tag=None, infos: dict=None):

Loading…
Cancel
Save