From fa25b637b0d257186a8399eb1c530a91f4252702 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期五, 14 四月 2023 15:44:50 +0800
Subject: [PATCH] remove some functions
---
funasr/models/joint_network.py | 5 ++---
1 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/funasr/models/joint_network.py b/funasr/models/joint_network.py
index 5cabdb4..ed827c4 100644
--- a/funasr/models/joint_network.py
+++ b/funasr/models/joint_network.py
@@ -2,7 +2,7 @@
import torch
-from funasr.modules.activation import get_activation
+from funasr.modules.nets_utils import get_activation
class JointNetwork(torch.nn.Module):
@@ -25,7 +25,6 @@
decoder_size: int,
joint_space_size: int = 256,
joint_activation_type: str = "tanh",
- **activation_parameters,
) -> None:
"""Construct a JointNetwork object."""
super().__init__()
@@ -36,7 +35,7 @@
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(
--
Gitblit v1.9.1