black formatted

This commit is contained in:
jinzr 2023-10-31 10:23:11 +08:00
parent fde8a2ff65
commit 72d947387d
2 changed files with 2 additions and 2 deletions

View File

@ -22,10 +22,11 @@
import argparse
import logging
import math
import pprint
from collections import defaultdict
from pathlib import Path
from typing import Dict, List, Optional, Tuple
import pprint
import k2
import sentencepiece as spm
import torch

View File

@ -253,7 +253,6 @@ class CausalSqueezeExcite1d(nn.Module):
return y
def forward(self, x: Tensor) -> Tensor:
assert len(x.shape) == 3, "Input is not a 3D tensor!"
y = self.exponential_moving_avg(x)
y = y.permute(0, 2, 1) # make channel last for squeeze op