testadmin
/
ParakeetEricRoss
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #142 from yt605155624/fastspeech2_cpu 

add fastspeech2 cpu train
This commit is contained in:
Hui Zhang 2021-08-17 07:36:31 -05:00 committed by GitHub
parent 24c5b3c1a2 cdd431e95b
commit 70b6ce64e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/fastspeech2/baker/train.py
View File

@ -65,13 +65,13 @@ def build_optimizers(model: nn.Layer, optim='adadelta',
def train_sp(args, config):
# decides device type and whether to run in parallel
# setup running environment correctly
if not paddle.is_compiled_with_cuda:
if not paddle.is_compiled_with_cuda():
paddle.set_device("cpu")
else:
paddle.set_device("gpu")
world_size = paddle.distributed.get_world_size()
if world_size > 1:
paddle.distributed.init_parallel_env()
world_size = paddle.distributed.get_world_size()
if world_size > 1:
paddle.distributed.init_parallel_env()
# set the random seed, it is a must for multiprocess training
seed_everything(config.seed)
@ -139,7 +139,7 @@ def train_sp(args, config):
odim = config.n_mels
model = FastSpeech2(idim=vocab_size, odim=odim, **config["model"])
if world_size > 1:
model = DataParallel(model) # TODO, do not use vocab size from config
model = DataParallel(model)
print("model done!")
optimizer = build_optimizers(model, **config["optimizer"])