dan
/
tacotron2
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

Merge pull request #4 from NVIDIA/mask-utils-0.4 

mask utils update for 0.4 cuda
master
Rafael Valle 6 years ago
committed by GitHub
parent
c141726a96 6fbba8ef0f
commit
2c545ac800
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      utils.py

+ 1
- 1
utils.py View File

@ -5,7 +5,7 @@ import torch
def get_mask_from_lengths(lengths): def get_mask_from_lengths(lengths):
max_len = torch.max(lengths) max_len = torch.max(lengths)
ids = torch.arange(0, max_len, out=torch.LongTensor(max_len)).cuda()
ids = torch.arange(0, max_len).long().cuda()
mask = (ids < lengths.unsqueeze(1)).byte() mask = (ids < lengths.unsqueeze(1)).byte()
return mask return mask

Write Preview
Loading…
Cancel
Save