From ace42e5043be992c3a324fc637ba0594f10ad3b6 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期五, 08 十二月 2023 16:39:38 +0800
Subject: [PATCH] fix loss normalization for ddp training
---
funasr/build_utils/build_sv_model.py | 2 --
1 files changed, 0 insertions(+), 2 deletions(-)
diff --git a/funasr/build_utils/build_sv_model.py b/funasr/build_utils/build_sv_model.py
index c0f1ae8..55df75a 100644
--- a/funasr/build_utils/build_sv_model.py
+++ b/funasr/build_utils/build_sv_model.py
@@ -1,7 +1,6 @@
import logging
import torch
-from typeguard import check_return_type
from funasr.layers.abs_normalize import AbsNormalize
from funasr.layers.global_mvn import GlobalMVN
@@ -254,5 +253,4 @@
if args.init is not None:
initialize(model, args.init)
- assert check_return_type(model)
return model
--
Gitblit v1.9.1