From e6a7bbe1ca6690faa23d29e22cb74a8d67c09ed3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 05 一月 2024 17:00:11 +0800
Subject: [PATCH] load_audio_text_image_video

---
 funasr/schedulers/__init__.py |   23 +++++++++++++++++++++++
 1 files changed, 23 insertions(+), 0 deletions(-)

diff --git a/funasr/schedulers/__init__.py b/funasr/schedulers/__init__.py
index e69de29..2ee3a9e 100644
--- a/funasr/schedulers/__init__.py
+++ b/funasr/schedulers/__init__.py
@@ -0,0 +1,23 @@
+import torch
+import torch.multiprocessing
+import torch.nn
+import torch.optim
+
+from funasr.schedulers.noam_lr import NoamLR
+from funasr.schedulers.tri_stage_scheduler import TriStageLR
+from funasr.schedulers.warmup_lr import WarmupLR
+
+scheduler_classes = dict(
+	ReduceLROnPlateau=torch.optim.lr_scheduler.ReduceLROnPlateau,
+	lambdalr=torch.optim.lr_scheduler.LambdaLR,
+	steplr=torch.optim.lr_scheduler.StepLR,
+	multisteplr=torch.optim.lr_scheduler.MultiStepLR,
+	exponentiallr=torch.optim.lr_scheduler.ExponentialLR,
+	CosineAnnealingLR=torch.optim.lr_scheduler.CosineAnnealingLR,
+	noamlr=NoamLR,
+	warmuplr=WarmupLR,
+	tri_stage=TriStageLR,
+	cycliclr=torch.optim.lr_scheduler.CyclicLR,
+	onecyclelr=torch.optim.lr_scheduler.OneCycleLR,
+	CosineAnnealingWarmRestarts=torch.optim.lr_scheduler.CosineAnnealingWarmRestarts,
+)
\ No newline at end of file

--
Gitblit v1.9.1