提交 82e04d78 编写于 作者: T tianhao zhang

fix trianer

上级 ed16f96a
...@@ -2,8 +2,8 @@ ...@@ -2,8 +2,8 @@
source path.sh source path.sh
set -e set -e
gpus=0,1,2,3 gpus=1
stage=0 stage=1
stop_stage=50 stop_stage=50
conf_path=conf/conformer.yaml conf_path=conf/conformer.yaml
ips= #xx.xx.xx.xx,xx.xx.xx.xx ips= #xx.xx.xx.xx,xx.xx.xx.xx
......
...@@ -19,6 +19,9 @@ from pathlib import Path ...@@ -19,6 +19,9 @@ from pathlib import Path
import paddle import paddle
from paddle import distributed as dist from paddle import distributed as dist
import pdb
pdb.set_trace()
dist.init_parallel_env()
from visualdl import LogWriter from visualdl import LogWriter
from paddlespeech.s2t.training.reporter import ObsScope from paddlespeech.s2t.training.reporter import ObsScope
...@@ -176,7 +179,7 @@ class Trainer(): ...@@ -176,7 +179,7 @@ class Trainer():
def init_parallel(self): def init_parallel(self):
"""Init environment for multiprocess training. """Init environment for multiprocess training.
""" """
dist.init_parallel_env() # dist.init_parallel_env()
@mp_tools.rank_zero_only @mp_tools.rank_zero_only
def save(self, tag=None, infos: dict=None): def save(self, tag=None, infos: dict=None):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册