From f2ed4b3856eaed8abe568e6904ffd8dc3a799f5f Mon Sep 17 00:00:00 2001
From: 彭震东 <zhendong.peng@qq.com>
Date: 星期一, 15 七月 2024 17:54:27 +0800
Subject: [PATCH] fix progress bar for batch_size (#1917)
---
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