From 46721dc84a6b4ecaac351022a47cc480c6dbd895 Mon Sep 17 00:00:00 2001
From: Thomas Young <hsyysy@163.com>
Date: 星期四, 05 十二月 2024 11:01:29 +0800
Subject: [PATCH] fix export error if using cpu device; fix model path (#2265)
---
funasr/optimizers/sgd.py | 2 --
1 files changed, 0 insertions(+), 2 deletions(-)
diff --git a/funasr/optimizers/sgd.py b/funasr/optimizers/sgd.py
index 3f0d3d1..fb7a3df 100644
--- a/funasr/optimizers/sgd.py
+++ b/funasr/optimizers/sgd.py
@@ -1,5 +1,4 @@
import torch
-from typeguard import check_argument_types
class SGD(torch.optim.SGD):
@@ -21,7 +20,6 @@
weight_decay: float = 0.0,
nesterov: bool = False,
):
- assert check_argument_types()
super().__init__(
params,
lr=lr,
--
Gitblit v1.9.1