From 19dcbc2d468a0831d52c8c52f023c4b8d0b5190c Mon Sep 17 00:00:00 2001 From: hnluo <haoneng.lhn@alibaba-inc.com> Date: 星期一, 05 六月 2023 17:38:24 +0800 Subject: [PATCH] Merge pull request #593 from alibaba-damo-academy/dev_lhn --- funasr/layers/abs_normalize.py | 14 ++++++++++++++ 1 files changed, 14 insertions(+), 0 deletions(-) diff --git a/funasr/layers/abs_normalize.py b/funasr/layers/abs_normalize.py new file mode 100644 index 0000000..4e617d0 --- /dev/null +++ b/funasr/layers/abs_normalize.py @@ -0,0 +1,14 @@ +from abc import ABC +from abc import abstractmethod +from typing import Tuple + +import torch + + +class AbsNormalize(torch.nn.Module, ABC): + @abstractmethod + def forward( + self, input: torch.Tensor, input_lengths: torch.Tensor = None + ) -> Tuple[torch.Tensor, torch.Tensor]: + # return output, output_lengths + raise NotImplementedError \ No newline at end of file -- Gitblit v1.9.1