From 27e8ba254adbe9f0df50b4969c5d2e66e0c7c0de Mon Sep 17 00:00:00 2001 From: jaluik <37829041+jaluik@users.noreply.github.com> Date: 星期二, 09 一月 2024 10:57:03 +0800 Subject: [PATCH] docs: update docker tag (#1227) --- funasr/schedulers/__init__.py | 23 ----------------------- 1 files changed, 0 insertions(+), 23 deletions(-) diff --git a/funasr/schedulers/__init__.py b/funasr/schedulers/__init__.py index 7bb8118..e69de29 100644 --- a/funasr/schedulers/__init__.py +++ b/funasr/schedulers/__init__.py @@ -1,23 +0,0 @@ -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_choices = 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