mirror of
https://github.com/shivammehta25/Matcha-TTS.git
synced 2026-02-05 18:29:19 +08:00
disable consider-using-generator
This commit is contained in:
@@ -234,9 +234,9 @@ class TextMelBatchCollate:
|
|||||||
|
|
||||||
def __call__(self, batch):
|
def __call__(self, batch):
|
||||||
B = len(batch)
|
B = len(batch)
|
||||||
y_max_length = max([item["y"].shape[-1] for item in batch])
|
y_max_length = max([item["y"].shape[-1] for item in batch]) # pylint: disable=consider-using-generator
|
||||||
y_max_length = fix_len_compatibility(y_max_length)
|
y_max_length = fix_len_compatibility(y_max_length)
|
||||||
x_max_length = max([item["x"].shape[-1] for item in batch])
|
x_max_length = max([item["x"].shape[-1] for item in batch]) # pylint: disable=consider-using-generator
|
||||||
n_feats = batch[0]["y"].shape[-2]
|
n_feats = batch[0]["y"].shape[-2]
|
||||||
|
|
||||||
y = torch.zeros((B, n_feats, y_max_length), dtype=torch.float32)
|
y = torch.zeros((B, n_feats, y_max_length), dtype=torch.float32)
|
||||||
|
|||||||
Reference in New Issue
Block a user