update

嘉渊

2023-04-24 6427c834dfd97b1f05c6659cdc7ccf010bf82fe1

 funasr/models/target_delay_transformer.py

@@ -6,13 +6,10 @@
import torch.nn as nn

from funasr.modules.embedding import SinusoidalPositionEncoder
#from funasr.models.encoder.transformer_encoder import TransformerEncoder as Encoder
from funasr.punctuation.sanm_encoder import SANMEncoder as Encoder
#from funasr.modules.mask import subsequent_n_mask
from funasr.train.abs_model import AbsPunctuation
from funasr.models.encoder.sanm_encoder import SANMEncoder as Encoder


class TargetDelayTransformer(AbsPunctuation):
class TargetDelayTransformer(torch.nn.Module):

    def __init__(
        self,