| | |
|---|
| | | import torch.nn as nn |
|---|
| | | import torch.nn.functional as F |
|---|
| | | |
|---|
| | | from funasr.utils.register import register_class, registry_tables |
|---|
| | | from funasr.register import tables |
|---|
| | | |
|---|
| | | class LinearTransform(nn.Module): |
|---|
| | | |
|---|
| | |
|---|
| | | fsmn_layers: no. of sequential fsmn layers |
|---|
| | | ''' |
|---|
| | | |
|---|
| | | @register_class("encoder_classes", "FSMN") |
|---|
| | | @tables.register("encoder_classes", "FSMN") |
|---|
| | | class FSMN(nn.Module): |
|---|
| | | def __init__( |
|---|
| | | self, |
|---|
| | |
|---|
| | | rstride: right stride |
|---|
| | | ''' |
|---|
| | | |
|---|
| | | @register_class("encoder_classes", "DFSMN") |
|---|
| | | @tables.register("encoder_classes", "DFSMN") |
|---|
| | | class DFSMN(nn.Module): |
|---|
| | | |
|---|
| | | def __init__(self, dimproj=64, dimlinear=128, lorder=20, rorder=1, lstride=1, rstride=1): |
|---|
