| | |
|---|
| | | |
|---|
| | | import torch |
|---|
| | | |
|---|
| | | from funasr.modules.activation import get_activation |
|---|
| | | from funasr.modules.nets_utils import get_activation |
|---|
| | | |
|---|
| | | |
|---|
| | | class JointNetwork(torch.nn.Module): |
|---|
| | |
|---|
| | | decoder_size: int, |
|---|
| | | joint_space_size: int = 256, |
|---|
| | | joint_activation_type: str = "tanh", |
|---|
| | | **activation_parameters, |
|---|
| | | ) -> None: |
|---|
| | | """Construct a JointNetwork object.""" |
|---|
| | | super().__init__() |
|---|
| | |
|---|
| | | self.lin_out = torch.nn.Linear(joint_space_size, output_size) |
|---|
| | | |
|---|
| | | self.joint_activation = get_activation( |
|---|
| | | joint_activation_type, **activation_parameters |
|---|
| | | joint_activation_type |
|---|
| | | ) |
|---|
| | | |
|---|
| | | def forward( |
|---|
