malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 09:42:34 +00:00
Code Issues Releases Wiki Activity

train.py: renaming function, removing dataparallel

Browse Source
This commit is contained in:
rafaelvalle
2018-11-27 18:04:12 -08:00
parent 3045ba125b
commit f06063f746
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@@ -84,9 +84,7 @@ def load_model(hparams):
model.decoder.attention_layer.score_mask_value = float(finfo('float16').min) model.decoder.attention_layer.score_mask_value = float(finfo('float16').min)
if hparams.distributed_run: if hparams.distributed_run:
model = DistributedDataParallel(model) model = apply_gradient_allreduce(model)
elif torch.cuda.device_count() > 1:
model = DataParallel(model)
return model return model