zixian commited on
Commit
6a6f110
1 Parent(s): 8810b16

Update monotonic_align/__init__.py

Browse files
Files changed (1) hide show
  1. monotonic_align/__init__.py +1 -1
monotonic_align/__init__.py CHANGED
@@ -16,4 +16,4 @@ def maximum_path(neg_cent, mask):
16
  t_t_max = mask.sum(1)[:, 0].data.cpu().numpy().astype(np.int32)
17
  t_s_max = mask.sum(2)[:, 0].data.cpu().numpy().astype(np.int32)
18
  maximum_path_c(path, neg_cent, t_t_max, t_s_max)
19
- return torch.from_numpy(path).to(device=device, dtype=dtype)
 
16
  t_t_max = mask.sum(1)[:, 0].data.cpu().numpy().astype(np.int32)
17
  t_s_max = mask.sum(2)[:, 0].data.cpu().numpy().astype(np.int32)
18
  maximum_path_c(path, neg_cent, t_t_max, t_s_max)
19
+ return torch.from_numpy(path).to(device=device, dtype=dtype)